finish generate_sequence.py
This commit is contained in:
17
core/dataset.py
Normal file
17
core/dataset.py
Normal 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
20
core/evaluation.py
Normal 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
|
||||
|
@@ -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__()
|
||||
|
Reference in New Issue
Block a user