python如何按照excel中csv文件的指定标签划分图像至指定类别文件夹

前言:请确保图像名称和csv文件中文件名一一对应,且csv中类别已经标明

#引入相关库

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

#输入表格所在路径+名称并读取,如D:/data/label.csv,文件后缀不要忘记
csv_path ='表格所在路径+名称'list=pd.read_csv(csv_path)
# print(list) #此命令可查看csv文件内容

path = ‘图片所在路径/.png’ #若图像格式为其他,则将后缀更改为其他

# list["File ID"] = input_path

list["File ID1"]=list["File_ID"] #File_ID为csv文件中图片文件名,含后缀,即与path路径的图片名完全一致

# # #
#label为0,1,2
for i in range(3):
    #即例如指定输出路径分别为D:/data/val/0,1,2
    output_path = '图片所在路径/val/'+str(i)
    #输出路径是否存在,若不存在则创建
    isExists = os.path.exists(output_path)
    if not isExists:
        os.makedirs(output_path)
    #label为csv文件中标签列的列名
    listnew = list[list["label"] == i]
    l = listnew["File ID1"].tolist()
    # 对于File ID中的每一个,执行移动指令
    for each in l:
        shutil.move(each, output_path)

#

亲测好用

  • 1
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值