使用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设计】。

相关文章

python机器学习案例教程——K最近邻算法的实现

K最近邻属于一种分类算法,他的解释最容易,近朱者赤,近墨者黑,我们想看一个人是什么样的,看他的朋友是什么样的就可以了。当然其他还牵着到,看哪方面和朋友比较接近(对象特征),怎样才算是跟朋...

儿童学习python的一些小技巧

以下是一些Python实用技巧和工具,希望能对大家有所帮助。 交换变量 x = 6 y = 5 x, y = y, x print x >>> 5 print...

python 输入一个数n,求n个数求乘或求和的实例

求和 try: while True: n=input() s=1 for x in raw_input().split(): s=s+int(x) prin...

浅谈python连续赋值可能引发的错误

今天写的代码片段: X = Y = [] .. X.append(x) Y.append(y) 其中x和y是读取的每一个数据的xy值,打算将其归入列表之后绘散点图,但是绘图出来却是...

浅谈Python数据类型之间的转换

Python数据类型之间的转换 函数 描述 int(x [,base])...