我有一个包含图像标签和路径的csv文件,我有另一个包含所有图像的文件夹,所以我想保存每个标签的图像在它自己的文件夹中,这里的csv看起来像,我感谢任何帮助enter image description here我找不到这个的任何代码
stszievb1#
您必须使用pandas阅读csv,os创建文件夹,shutil复制文件。
pandas
os
shutil
import os import shutil import pandas as pd # read the file csv_file = pd.read_csv('file.csv', dtype=str) # create the folders labels = csv_file['label'] for label in labels: os.makedirs(label, exist_ok=True) # iterate rows and copy images for _, row in csv_file.iterrows(): label = row['label'] path = row['path'] img_name = os.path.split(path)[-1] new_path = os.path.join(label, img_name) shutil.copy(path, new_path)
1条答案
按热度按时间stszievb1#
您必须使用
pandas
阅读csv,os
创建文件夹,shutil
复制文件。