File size: 401 Bytes
02c5426 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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')
|