2024-08-22 20:27:21 +08:00

17 lines
491 B
Python

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