templates/tensorf/dataLoader/__init__.py

from .llff import LLFFDataset
from .blender import BlenderDataset
from .nsvf import NSVF
from .tankstemple import TanksTempleDataset
from .your_own_data import YourOwnDataset

dataset_dict = {'blender': BlenderDataset,
                'llff': LLFFDataset,
                'tankstemple': TanksTempleDataset,
                'nsvf': NSVF,
                'own_data': YourOwnDataset}