| import glob | |
| import os | |
| import numpy as np | |
| import pickle | |
| import sys | |
| import tqdm | |
| import shutil | |
| from skimage import io | |
| pre_path = r'H:\DataSet\SceneCls\UCMerced_LandUse\UCMerced_LandUse\Images' | |
| sub_folder_list = glob.glob(pre_path +'/*') | |
| all_data_list = [] | |
| for sub_folder in sub_folder_list: | |
| img_list = glob.glob(sub_folder+'/*') | |
| all_data_list += img_list | |
| with open(pre_path+f'/../all_img_list.txt', 'w') as f: | |
| for file in tqdm.tqdm(all_data_list): | |
| img = io.imread(file, as_gray=True) | |
| if 0 < img.shape[0]: | |
| file_name = os.path.basename(os.path.dirname(file)) + '/' + os.path.basename(file) | |
| gt_label = os.path.basename(os.path.dirname(file)) | |
| f.write(file_name+' '+gt_label+'\n') | |