图片按照csv文件中的标签保存在不同文件夹下

最新推荐文章于 2025-07-10 12:11:45 发布

原创最新推荐文章于 2025-07-10 12:11:45 发布 · 1.3k 阅读

11 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

机器学习

当前文章被收录于：

机器学习

5 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

该博客介绍了一个Python脚本，该脚本根据CSV文件中的标签将图像文件分类到不同的文件夹中。它首先读取包含图像标签的CSV文件，然后遍历标签列表，为每个类别创建相应的文件夹，并将对应标签的图像复制到对应的文件夹。这个过程对于数据预处理和机器学习项目非常有用。

图片按照csv文件中的标签保存在不同文件夹下，文件夹以标签名字命名。

原始文件路径： data-train

-test

-train.csv

filename	label
0.jpg	plastic
1.jpg	paper

import pandas as pd
import os
import shutil #用于移动文件

#打开表格文件并读取
f=open("./refuse_data/train.csv","rb") #输入表格所在路径+名称
list=pd.read_csv(f)

#进行分类
for i in ['paper','glass','plastic','cardboard','metal','trash']:
    os.mkdir(i)
    listnew=list[list["label"]==i]
    l=listnew["filename"].tolist()
    j=str(i)
    for each in l:
        print(each,j)
        shutil.copy('./refuse_data/train/'+each,j)