网络编程
位置:首页>> 网络编程>> Python编程>> 使用python将图片按标签分入不同文件夹的方法

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

作者:翻滚的老鼠屎  发布时间:2021-04-14 05:34:54 

标签:python,图片,标签,文件夹

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

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

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

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

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


#引入相关库

import pandas as pd
import os
import 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将图片按标签分入不同文件夹的方法

来源:https://blog.csdn.net/LaoChengZier/article/details/80254393

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com