clean/video/fakestormer/datasets/builder.py

# -*- coding: utf-8 -*-
import os
import sys
from typing import Any, Dict, Optional

if not os.getcwd() in sys.path:
    sys.path.append(os.getcwd())

from register.register import Registry, build_from_cfg

PIPELINES = Registry("Pipeline", build_func=build_from_cfg)
DATASETS = Registry("Dataset", build_func=build_from_cfg)


def build_pipeline(
    cfg,
    pipeline: Registry,
    build_func=build_from_cfg,
    default_args: Optional[Dict] = None,
) -> Any:
    return build_func(cfg, pipeline, default_args)


def build_dataset(
    cfg,
    dataset: Registry,
    build_func=build_from_cfg,
    default_args: Optional[Dict] = None,
) -> Any:
    return build_func(cfg, dataset, default_args)