finish generate_sequence.py

This commit is contained in:
hofee
2024-08-22 20:27:21 +08:00
parent ff3e89b17c
commit 7cd1954a25
7 changed files with 121 additions and 57 deletions

17
core/dataset.py Normal file
View File

@@ -0,0 +1,17 @@
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