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

yipeiwu_com5年前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 排序算法总结及实例详解

python 排序算法总结及实例详解

总结了一下常见集中排序的算法 归并排序 归并排序也称合并排序,是分治法的典型应用。分治思想是将每个问题分解成个个小问题,将每个小问题解决,然后合并。 具体的归并排序就是,将一组无序数...

利用一个简单的例子窥探CPython内核的运行机制

我最近花了一些时间在探索CPython,并且我想要在这里分享我的一些冒险经历。Allison Kaptur的excellent guide to getting started with...

python numpy数组的索引和切片的操作方法

NumPy - 简介 NumPy 是一个 Python 包。 它代表 “Numeric Python”。 它是一个由多维数组对象和用于处理数组的例程集合组成的库。 Numeric,即 N...

使用Python进行防病毒免杀解析

使用Python进行防病毒免杀解析

很多渗透工具都提供了权限维持的能力,如Metasploit、Empire和Cobalt Strike,但是都会被防病毒软件检测到这种恶意行为。在探讨一个权限维持技巧的时候,似乎越来越多的...

解决python matplotlib imshow无法显示的问题

实例如下所示: import matplotlib.pyplot as plt plt.imshow(img) #控制台打印出图像对象的信息,而图像没有显示 解决方法: #引入pyl...