使用python将图片按标签分入不同文件夹的方法

时间:2021-05-22

给定图像集如下,所有类别的图片均在一个文件夹内:

给定与图片名相匹配的表格,声明每张图片对应的类别(共有20个类别):

那么,如何根据表格中所给的类别将图片分入对应的文件夹内呢?以我的情况为例,我想将图片分为20类(CATEGORY_ID有0-19共20类),可利用下面的代码进行分类(经细心网友指正,代码已做出修改)。

#引入相关库import pandas as pdimport osimport shutil #用于移动文件#打开表格文件并读取f=open("list.csv","rb") #输入表格所在路径+名称list=pd.read_csv(f)list["FILE_ID_JPG"]=".jpg" #建立图片名与类别相对应list["FILE_ID1"]=list["FILE_ID"]+list["FILE_ID_JPG"] #建立图片名与类别相对应#创建文件夹for i in range(20): os.mkdir(str(i))#进行分类for i in range(0,20): listnew=list[list["CATEGORY_ID"]==i] l=listnew["FILE_ID1"].tolist() j=str(i) for each in l: shutil.move(each,j)

大功告成!

以上这篇使用python将图片按标签分入不同文件夹的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章