| 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 |