| import glob | |
| import os | |
| import numpy as np | |
| import pickle | |
| import sys | |
| import tqdm | |
| import shutil | |
| pre_path = r'H:\DataSet\SceneCls\UCMerced_LandUse\UCMerced_LandUse\Images' | |
| sub_folder_list = glob.glob(pre_path +'/*') | |
| with open(pre_path+f'/../class_names.txt', 'w') as f: | |
| for sub_folder in sub_folder_list: | |
| sub_folder_name = os.path.basename(sub_folder) | |
| f.write(sub_folder_name+'\n') | |