Python实现按标签将图片自动分类至不同文件夹
发布时间：2026-10-13 |
5人浏览过
本文介绍如何利用Python脚本，结合pandas读取包含图片类别信息的CSV表格，并通过os和shutil库自动创建文件夹及移动图片。该方法适用于需要将大量图片按预设标签（如0-19类）批量整理至独立目录的场景，代码已优化修正，可直接参考使用。
假设现有如下图像数据集，所有类别的图片目前均存储在同一文件夹中：
存在一份与图片文件名相匹配的表格，其中声明了每张图片所属的类别（共包含20个类别）：
那么，如何依据表格中提供的类别信息，将图片分别归入对应的文件夹呢？以我的具体场景为例，目标是将图片划分为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将图片按标签分入不同文件夹的方法即为分享的全部内容，希望能为读者提供有益参考，同时也欢迎大家继续支持本站。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
