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

相关文章

TensorFlow损失函数专题详解

TensorFlow损失函数专题详解

一、分类问题损失函数——交叉熵(crossentropy) 交叉熵刻画了两个概率分布之间的距离,是分类问题中使用广泛的损失函数。给定两个概率分布p和q,交叉熵刻画的是两个概率分布之间的距...

Python的Asyncore异步Socket模块及实现端口转发的例子

Asyncore模块提供了以异步的方式写入套接字服务客户端和服务器的基础结构。 只有两种方式使一个程序在单处理器上实现“同时做不止一件事”。多线程编程是最简单和最流行的方式,但是有另一种...

python读取json文件并将数据插入到mongodb的方法

本文实例讲述了python读取json文件并将数据插入到mongodb的方法。分享给大家供大家参考。具体实现方法如下: #coding=utf-8 import sunburnt i...

python每隔N秒运行指定函数的方法

本文实例讲述了python每隔N秒运行指定函数的方法。分享给大家供大家参考。具体如下: 这是一个类似定时器的效果,每隔指定的秒数运行指定的函数,采用线程实现,代码简单实用。 复制代码 代...

利用QT写一个极简单的图形化Python闹钟程序

今天我们讲一下用Python写的GUI小程序。一个小闹钟(只是屏幕提示,没有声音哦) 让我们先介绍这个闹钟如何奇葩。 需要通过命令行启动。 没有标题栏。 没菜单。 甚至没有关闭按钮。 没...