fix output of inference

app.py

@@ -96,12 +96,12 @@ def run_model(target_dir, model) -> dict:
     all_camera_pose = []
     
     for res in output.ress:
-        all_pts3d.append(res['pts3d_in_other_view'])
-        all_conf.append(res['conf'])
-        all_depth.append(res['depth'])
-        all_depth_conf.append(res['depth_conf'])
-        all_camera_pose.append(res['camera_pose'])
-    
+        all_pts3d.append(res['pts3d_in_other_view'].squeeze(0))
+        all_conf.append(res['conf'].squeeze(0))
+        all_depth.append(res['depth'].squeeze(0))
+        all_depth_conf.append(res['depth_conf'].squeeze(0))
+        all_camera_pose.append(res['camera_pose'].squeeze(0))
+
     predictions["world_points"] = torch.stack(all_pts3d, dim=0)  # (S, H, W, 3)
     predictions["world_points_conf"] = torch.stack(all_conf, dim=0)  # (S, H, W)
     predictions["depth"] = torch.stack(all_depth, dim=0)  # (S, H, W, 1)