python-图像分类数据集 给三级目录里的文件打标签 并且标签根据第二级级目录自动生成

​# -*- coding: utf-8 -*-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 = '文件地址'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 = '保存txt地址'creat_txtfile(output_path, file_list)if __name__ == '__main__':main()原文链接Add files via upload · leeleeleeleee/creat_txt_dataset@8f4a28c ()​