upd inference

2025-01-05 23:57:33 +08:00
parent 9c2625b11e
commit dec67e8255
4 changed files with 245 additions and 77 deletions
--- a/core/pipeline.py
+++ b/core/pipeline.py
@@ -75,6 +75,8 @@ class NBVReconstructionPipeline(nn.Module):

    def forward_test(self, data):
        main_feat = self.get_main_feat(data)
+        repeat_num = data.get("repeat_num", 100)
+        main_feat = main_feat.repeat(repeat_num, 1)
        estimated_delta_rot_9d, in_process_sample = self.view_finder.next_best_view(
            main_feat
        )