Python根据文件名批量转移图片

来源:互联网 发布:淘宝店的发展前景 编辑:程序博客网 时间:2024/05/22 06:21

下面是在深度学习数据集处理过程中可能会用到的一个小程序,帮助我们根据图片文件的名字来分开图片:

import osimport shutilpath_img='读取图片的路径'ls = os.listdir(path_img)print(len(ls))for i in ls:    if i.find('查找的关键词')!=-1:       shutil.move(path_img+'/'+i,"输出保存的路径"+i)

——————————————使用示例———————————————

例如我有如下的一个图片集,可以看到男性的图片都包含有“testnan”,女性的图片都包含“testnv”,现在想要把两个数据集分开独立的打标签,就可以用上面的小程序了。

这里写图片描述

import osimport shutilpath_img='C:\\Users\\chriszhang\\Desktop\\gender\\test'ls = os.listdir(path_img)print(len(ls))for i in ls:    if i.find('testnan')!=-1:        shutil.move(path_img+'/'+i,"C:/Users/chriszhang/Desktop/male/"+i)

如上就可以将我桌面上gender/test文件夹中的男性图片转到桌面上的male文件夹中了。