update pbnbv

2025-05-19 16:32:04 +08:00
parent b98753bfbb
commit cd436605a5
25 changed files with 1228 additions and 178 deletions
--- a/core/ab_mlp_pipeline.py
+++ b/core/ab_mlp_pipeline.py
@@ -45,6 +45,8 @@ class NBVReconstructionMLPPipeline(nn.Module):
    
    def forward_test(self,data):
        main_feat = self.get_main_feat(data)
+        repeat_num = data.get("repeat_num", 50)
+        main_feat = main_feat.repeat(repeat_num, 1)
        estimated_delta_rot_9d, _ = self.view_finder.next_best_view(main_feat)
        result = {
            "pred_pose_9d": estimated_delta_rot_9d,