Python提取测试集训练集图片路径txt-解释版本

import os
# 修改path_路径，为train的上一级文件夹地址
path_1='dataset/hymenoptera_data'

path_1_name=os.listdir(path_1)      # 将hymenoptera_data目录下子文件名，列出
len_path_1=len(path_1_name)         # list长度值，作为索引值

for idx in range(len_path_1):        # 索引遍历train和test文件夹
    file_label=path_1_name[idx]      # 将列表中的每一个文件名取出，比如当file_label=train时
    dir_path = path_1+'/'+file_label # 构建dir_path：dataset/hymenoptera_data/train
    dir_name=os.listdir(dir_path)    # 列表中为train目录下，所有文件名
    x=len(dir_name)                  # 长度作为索引值
    for file_name in range(x):       # 遍历train目录下子文件：ants,bees
        label_name=dir_name[file_name] # 取出类别名，ants，bees
        dir_path_1=os.path.join(dir_path,label_name) # dataset/hymenoptera_data/train\ants
        filename_list=os.listdir(dir_path_1)    # 列表：ataset/hymenoptera_data/train\ants下所有的图片名称
        lenx=len(filename_list)                 # 长度值作为索引
        save_path = path_1+'/'+file_label+'/'+'{}'.format(file_label)+'.txt' # txt文件保存路径
        for i in range(lenx):                   # 遍历所有图片
            pic_path=os.path.join(dir_path,label_name,filename_list[i]) # 提取单张图片地址
            with open(save_path, 'a') as f:  # 打开文件，追加写入
                f.write(pic_path)
                f.write('\n')                # 按行写入