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

20
core/evaluation.py Normal file
View File

@@ -0,0 +1,20 @@
import torch
import PytorchBoot.stereotype as stereotype
@stereotype.evaluation_method("delta_pose_diff", comment="unfinished")
class DeltaPoseDiff:
def __init__(self, config):
pass
def evaluate(self, output_list, data_list):
return
@stereotype.evaluation_method("coverage_rate_increase",comment="unfinished")
class ConverageRateIncrease:
def __init__(self, config):
pass
def evaluate(self, output_list, data_list):
return

View File

@@ -5,7 +5,7 @@ import PytorchBoot.stereotype as stereotype
from PytorchBoot.factory.component_factory import ComponentFactory
from PytorchBoot.utils import Log
@stereotype.pipeline("nbv_reconstruction_pipeline")
@stereotype.pipeline("nbv_reconstruction_pipeline", comment="should be tested")
class NBVReconstructionPipeline(nn.Module):
def __init__(self, config):
super(NBVReconstructionPipeline, self).__init__()