2020-11-09数据集读取

最新推荐文章于 2026-08-22 10:48:54 发布

原创最新推荐文章于 2026-08-22 10:48:54 发布 · 269 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

当前文章被以下社区和专栏收录：

该脚本根据CSV文件中的数据将图片从原始路径移动到目标路径下的相应类别文件夹中。使用pandas读取CSV文件，并检查目标路径下是否存在对应的类别文件夹，如果不存在则创建。然后将指定路径的图片移动到相应的类别文件夹。

import csv
import shutil
import os
import pandas as pd
target_path = './data_test/'
original_path = './data/input/ButterflyClassification/'
with open('train.csv',"rt", encoding="utf-8") as csvfile:
    df = pd.read_csv('./data/input/ButterflyClassification/train.csv')
    for i in range(len(df)):
        if os.path.exists(target_path+df['label'][i]) :
            full_path = original_path + df['image_path'][i]
            shutil.move(full_path,target_path +df['label'][i] +'/')
        else :
            os.makedirs(target_path+df['label'][i])
            full_path = original_path + df['image_path'][i]
            shutil.move(full_path,target_path + df['label'][i] +'/')