IGEV/IGEV-MVS/datasets/__init__.py
2023-03-20 19:52:04 +08:00

9 lines
271 B
Python

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")