IGEV/IGEV-MVS/datasets/__init__.py

9 lines
271 B
Python
Raw Normal View History

2023-03-20 19:52:04 +08:00
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")