import os# 分三级目录，如A/B/a.jpg
# input_path为一级目录；
#
def creat_filelist(input_path, classes):# 创建三级目录# index 一定是str类型，不可以为intdir_image1 = []  # 二级目录file_list = []  # 三级目录for index, name in enumerate(classes):print('index', index)index_str = str(index)dir_image1_temp = input_path + '/' + name + '/'for dir2 in os.listdir(dir_image1_temp):dir_image2_temp = dir_image1_temp + '/' + dir2 + ' ' + index_str# dir_image2_temp1 = dir_image2_temp.join(' ')# dir_image2_temp2 = dir_image2_temp.join(index)file_list.append(dir_image2_temp)return dir_image1, file_listdef creat_txtfile(output_path, file_list):with open(output_path, 'w') as f:for list in file_list:print(list)f.write(str(list) + '\n')def main():dir_image0 = './data2'dir_image1 = os.listdir(dir_image0)classes = dir_image1print(classes)dir_list, file_list = creat_filelist(dir_image0, classes)# print(file_list[0:3])output_path = './train_data1.txt'creat_txtfile(output_path, file_list)if __name__ == '__main__':main()