__init__.py 271 字节
Newer Older
Gangwei Xu's avatar
Gangwei Xu 已提交
1
2
3
4
5
6
7
8
import importlib


# find the dataset definition by name, for example dtu_yao (dtu_yao.py)
def find_dataset_def(dataset_name):
    module_name = 'datasets.{}'.format(dataset_name)
    module = importlib.import_module(module_name)
    return getattr(module, "MVSDataset")