import os
data_path = []
data_name = []
for root, dirs, files in os.walk('train'):
# 变量指定目录文件列表
for image_file in files:
image_path = os.path.join(root, image_file)
data_path.append(image_path)
data_name.append(image_file.split('.')[0])
train文件夹下为图像数据集:
data_name:
data_path: