tools/video_dataset_select_one.py

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