1.读取文件夹所有图片并保存txt

import sys    
import os
import random 

data_base_dir = r"E:\DATA\UCMerced_LandUse-2\images" 
file_list = [] 
write_file_name = r"E:\DATA\UCMerced_LandUse-2\label.txt"
f=open(write_file_name, 'w')
i=0
for file in os.listdir(data_base_dir): 
    
    name =data_base_dir+'\'+ file
    for img_name in os.listdir(name):
        label=file+'/'+img_name+' '+str(i)+'\n'
        f.write(label)
    i=i+1
f.close()
import sys    
import os