File size: 741 Bytes
02c5426 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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')
|