下面是在深度学习数据集处理过程中可能会用到的一个小程序,帮助我们根据图片文件的名字来分开图片:
1
2
3
4
5
6
7
8
|
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”,现在想要把两个数据集分开独立的打标签,就可以用上面的小程序了。
1
2
3
4
5
6
7
8
|
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根据文件名批量转移图片的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/chris_zhangrx/article/details/78679895