Use height to calculate kitti centroid (#167)

Closes: #163
ch-sa · Aug 18, 2024 · 009aecd · 009aecd
1 parent dcfe94f
commit 009aecd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/labelCloud/io/labels/kitti.py b/labelCloud/io/labels/kitti.py
@@ -144,23 +144,23 @@ def export_labels(self, bboxes: List[BBox], pcd_path: Path) -> None:
         for bbox in bboxes:
             obj_type = bbox.get_classname()
             centroid = bbox.get_center()
-            dimensions = bbox.get_dimensions()
+            length, width, height = bbox.get_dimensions()
 
             # invert sequence to height, width, length
-            dimensions = dimensions[2], dimensions[1], dimensions[0]
+            dimensions = height, width, length
 
             if self.transformed:
                 try:
                     self._get_transforms(pcd_path)
-                except CalibrationFileNotFound as exc:
+                except CalibrationFileNotFound:
                     logging.exception("Calibration file not found")
                     logging.warning("Skipping writing of labels for this point cloud")
                     return
 
                 centroid = (
                     centroid[0],
                     centroid[1],
-                    centroid[2] - dimensions[2] / 2,
+                    centroid[2] - height / 2,
                 )  # centroid in KITTI located on bottom face of bbox
                 xyz1 = np.insert(np.asarray(centroid), 3, values=[1])
                 xyz1 = self.T_v2c @ xyz1