【生成数据集csv文件，第一列为路径，第二列为label】

# 导入相关库
import csv
import random
import os
import glob
root = '路径'
# 对数据地址中的文件夹进行遍历，将类名存放于列表names中
names = os.listdir(root)
# 存放图像地址
images = []

# 遍历类名
for name in names:
    images += glob.glob(os.path.join(root, name, '*.png'))

# 对csv文件进行写操作，如果没有csv文件会自动创建
with open(os.path.join(root, '保存的csv文件.csv'), mode='w', newline='') as f:
    writer = csv.writer(f)
    for img in images:
        name = img.split(os.sep)[-2] #label
        writer.writerow([img, name])