2022.08.30 19:22

2gunsu · Aug 30, 2022 · d40ebb0 · d40ebb0
1 parent f5d69aa
commit d40ebb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/dense_heads/monocon_heads.py b/model/dense_heads/monocon_heads.py
@@ -459,7 +459,7 @@ def decode_heatmap(self,
 
         depth = depth_pred[:, :, 0:1]                                                           # (B, K, 1)
         center3d = torch.cat([center2d, depth], dim=-1)                                         # (B, K, 3)
-        center3d = self.convert_pts2D_to_pts3D(center3d, batched_calib=data_dict['calib'])  # (B, K, 3)
+        center3d = self.convert_pts2D_to_pts3D(center3d, batched_calib=data_dict['calib'])      # (B, K, 3)
 
         dim = transpose_and_gather_feat(pred_dict['dim_pred'], indices)
         bboxes_3d = torch.cat([center3d, dim, rot_y], dim=-1)