Spaces:
Running
on
T4
Running
on
T4
File size: 596 Bytes
561c629 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os, shutil
dir = "../datasets/VideoLQ"
store_dir = "../datasets/VideoLQ_select_one"
if os.path.exists(store_dir):
shutil.rmtree(store_dir)
os.makedirs(store_dir)
search_idx = 0
for sub_folder_name in sorted(os.listdir(dir)):
sub_folder_dir = os.path.join(dir, sub_folder_name)
for idx, img_name in enumerate(sorted(os.listdir(sub_folder_dir))):
if idx != search_idx:
continue
img_path = os.path.join(sub_folder_dir, img_name)
target_path = os.path.join(store_dir, img_name)
shutil.copy(img_path, target_path)
search_idx += 1 |