服务器之家

服务器之家 > 正文

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

时间:2021-04-26 00:43     来源/作者:翻滚的老鼠屎

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

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

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

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

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

?
1
2
3
4
5
#引入相关库
 
import pandas as pd
import os
import shutil #用于移动文件
?
1
2
3
4
5
#打开表格文件并读取
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"] #建立图片名与类别相对应
?
1
2
3
#创建文件夹
for i in range(20):
 os.mkdir(str(i))
?
1
2
3
4
5
6
7
#进行分类
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将图片按标签分入不同文件夹的方法

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

原文链接:https://blog.csdn.net/LaoChengZier/article/details/80254393

相关文章

热门资讯

2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全 2019-12-26
yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总 2020-11-13
Intellij idea2020永久破解,亲测可用!!!
Intellij idea2020永久破解,亲测可用!!! 2020-07-29
返回顶部