时间:2022-08-09 08:33:35 | 栏目:Python代码 | 点击:次
下面是在深度学习数据集处理过程中可能会用到的一个小程序,帮助我们根据图片文件的名字来分开图片:
import os import shutil path_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 os import shutil path_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文件夹中了。