selection_dataset_image.py

import shutil
import os

SELECTION_DIR_NAME = 'face_dataset-2022-1-28'

for dir_dataset in os.listdir(os.path.join(os.getcwd(), 'face_dataset')):

    if not os.path.isdir(os.path.join(os.getcwd(), 'face_dataset', dir_dataset)):
        continue

    if not os.path.isdir(os.path.join(os.getcwd(), SELECTION_DIR_NAME, dir_dataset)):
        os.mkdir(os.path.join(os.getcwd(), SELECTION_DIR_NAME, dir_dataset))
        print("Create dir: " + dir_dataset)

    print(len(os.listdir(os.path.join(os.getcwd(), 'face_dataset', dir_dataset))))

    for file in os.listdir(os.path.join(os.getcwd(), 'face_dataset', dir_dataset)) \
            [::int((len(os.listdir(os.path.join(os.getcwd(), 'face_dataset', dir_dataset)))) / 30) + 1]:

        if '.jpg' not in file:
            continue

        print(dir_dataset, file)

        shutil.copy2(os.path.join(os.getcwd(), 'face_dataset', dir_dataset, file),
                     os.path.join(os.getcwd(), SELECTION_DIR_NAME, dir_dataset, file))