我有一个用于分割项目的数据集。数据集有 2 个文件夹，包含 train 文件夹和 annotated_train_data 文件夹，两个文件夹都有图像。输入和标签都是图像。我想制作这个数据集的 csv 文件以输入神经网络。制作csv文件最简单有效的方法是什么？
./images
./images/train
./images/train/label1.jpg
./images/train/label2.jpg
./images/train/label3.jpg
./images/annotated_train_data
./images/annotated_train_data/label1.jpg
./images/annotated_train_data/label2.jpg
./images/annotated_train_data/label3.jpg
import os
import pandas as pd
BASE_DIR = 'images/'
train_folder = BASE_DIR+'train/'
train_annotation = BASE_DIR+'annotated_train_data/'
files_in_train = sorted(os.listdir(train_folder))
files_in_annotated = sorted(os.listdir(train_annotation))
images=[i for i in files_in_train if i in files_in_annotated]
df = pd.DataFrame()
df['images']=[train_folder+str(x) for x in images]
df['labels']=[train_annotation+str(x) for x in images]
pd.to_csv('files_path.csv', header=None)
试试这个。