当前位置:首页 >> 脚本专栏

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”,现在想要把两个数据集分开独立的打标签,就可以用上面的小程序了。

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

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文件夹中了。

以上这篇Python根据文件名批量转移图片的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。