csv 如何分割图像取决于它的标签,使每个标签将有它的图像的文件夹

j2cgzkjk  于 2022-12-06  发布在  其他
关注(0)|答案(1)|浏览(123)

我有一个包含图像标签和路径的csv文件,我有另一个包含所有图像的文件夹,所以我想保存每个标签的图像在它自己的文件夹中,这里的csv看起来像,我感谢任何帮助
enter image description here
我找不到这个的任何代码

stszievb

stszievb1#

您必须使用pandas阅读csv,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)

相关问题