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

yipeiwu_com6年前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将图片按标签分入不同文件夹

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

相关文章

Python3远程监控程序的实现方法

简述 一开始觉得这个很有趣,然后就想来做一个来玩一下 使用语言: Python3 使用工具:opencv视频监控 + socket数据传输技术 程序检验: 这里我考虑了一下,发现还是没有...

Python使用import导入本地脚本及导入模块的技巧总结

本文实例讲述了Python使用import导入本地脚本及导入模块的技巧。分享给大家供大家参考,具体如下: 导入本地脚本 import 如果你要导入的 Python 脚本与当前脚本位于同一...

对python中的高效迭代器函数详解

对python中的高效迭代器函数详解

python中内置的库中有个itertools,可以满足我们在编程中绝大多数需要迭代的场合,当然也可以自己造轮子,但是有现成的好用的轮子不妨也学习一下,看哪个用的顺手~ 首先还是要先im...

利用nohup来开启python文件的方法

python文件可以直接使用命令python xxx.py来启动文件,但是这样会有一个弊端,就是关闭ssh连接,python文件就会自动的进行停止。 所以需要使用利用nohup来开启py...

说说如何遍历Python列表的方法示例

说说如何遍历Python列表的方法示例

如果需要对列表中的每个元素执行相同操作,这时就需要遍历列表的所有元素。 books=['半生缘','往事并不如烟','心是孤独的猎手'] for book in books: p...