From 96fb34be0923e0e3a25302a3fe194243e0366741 Mon Sep 17 00:00:00 2001 From: hofee Date: Wed, 21 Aug 2024 17:59:42 +0800 Subject: [PATCH] change pipeline's name --- core/pipeline.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/pipeline.py b/core/pipeline.py index b7256f5..ceb5c03 100644 --- a/core/pipeline.py +++ b/core/pipeline.py @@ -5,9 +5,9 @@ import PytorchBoot.stereotype as stereotype from PytorchBoot.factory.component_factory import ComponentFactory @stereotype.pipeline("nbv_reconstruction_pipeline") -class ViewFinderPipeline(nn.Module): +class NBVReconstructionPipeline(nn.Module): def __init__(self, config): - super(ViewFinderPipeline, self).__init__() + super(NBVReconstructionPipeline, self).__init__() self.config = config self.pts_encoder = ComponentFactory.create(namespace.Stereotype.MODULE, config["pts_encoder"]) self.pose_encoder = ComponentFactory.create(namespace.Stereotype.MODULE, config["pose_encoder"]) @@ -24,6 +24,6 @@ class ViewFinderPipeline(nn.Module): pts_feat_list.append(self.pts_encoder.encode_points(pts)) pose_feat_list.append(self.pose_encoder.encode_pose(pose)) seq_feat = self.seq_encoder.encode_sequence(pts_feat_list, pose_feat_list) - output['next_best_view'] = self.view_finder.next_best_view(seq_feat) + output['estimated_score'] = self.view_finder.next_best_view(seq_feat) return output \ No newline at end of file