import pandas as pd import os import shutil # 读取文件 file = open("D:\\Project\\test\\picture\\data.csv", "rb") list = pd.read_csv(file) list["FILE_ID"] = list["ID"] root = "D:\\Project\\test\\picture" # 创建文件夹,进行分类 for i in range(9): label_dir = os.path.join(root,"{:1}".format(i)) if not os.path.exists(label_dir): os.mkdir(label_dir) listnew = list[list["label"] == i] print(listnew) list_image = listnew["FILE_ID"].tolist() print(list_image) for img in list_image: print(img) shutil.copy("D:\\Project\\test\\picture\\img\\" + img, label_dir)
按标注将图片分类
最新推荐文章于 2023-10-14 14:49:36 发布