prepare dataset

prepare.py

import os
from shutil import copyfile


def _make_sure_dir_exist(path):
    if not os.path.isdir(path):
        os.mkdir(path)


def refactor_dataset_folder(root_path, classes_map, img_path_root=None):
    if not os.path.isdir(root_path):
        raise NotADirectoryError(root_path + " do not exists")
    for c in classes_map:
        class_root = os.path.join(root_path, c)
        _make_sure_dir_exist(class_root)
        for img_path in classes_map[c]:
            if img_path_root is not None:
                real_img_path = os.path.join(img_path_root, img_path)
            else:
                real_img_path = img_path

            copyfile(real_img_path, os.path.join(class_root, os.path.basename(img_path)))