from PytorchBoot.dataset import BaseDataset import PytorchBoot.stereotype as stereotype @stereotype.dataset("nbv_reconstruction_dataset", comment="unfinished") class NBVReconstructionDataset(BaseDataset): def __init__(self, config): super(NBVReconstructionDataset, self).__init__(config) self.config = config def get_datalist(self): pass def load_view(path): pass def load_data_item(self, idx): pass