python-根据csv数据标签给图片分类存储
python-根据csv数据标签给图片分类存储
·
import os
import shutil
import pandas as pd
import random
# 打开表格文件并读取
f = open("E:train1/trainLabels.csv", "rb") # 打开csv文件
list = pd.read_csv(f) # 这句不能少
# print(list)
# 创建文件夹
for i in range(3): # "6"指的是0-5总共6个类别
if not os.path.exists('E:/train1/' + str(i)): # 最后一个 / 不要漏
os.mkdir('E:/train1/' + str(i))
# 进行分类
for i in range(3):
listnew = list[list["level"] == i] # 对应csv文件标签 那一栏的标题
l = listnew["image"].tolist() # 对应csv文件图片那一栏的标题
j = str(i)
for each in l:
shutil.move('E:/train1/train/' + each +'.jpeg', 'E:/train1/' + j)
print("完成")
亲测可以跑通执行
更多推荐


所有评论(0)