fix: three dimensional points for get body points

UCSD-E4E · Sep 15, 2024 · 973e3e0 · 973e3e0
1 parent cdc3f85
commit 973e3e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/pyfishsensedev/image/pdf.py b/pyfishsensedev/image/pdf.py
@@ -33,5 +33,5 @@ def height(self) -> int:
 
         return height
 
-    def get_points_body_space(self, points: np.ndarray) -> np.ndarray:
+    def get_physical_measurements(self, points: np.ndarray) -> np.ndarray:
         return (points / float(Pdf.DPI)) * INCH_TO_M
diff --git a/pyfishsensedev/plane_detector/slate_detector.py b/pyfishsensedev/plane_detector/slate_detector.py
@@ -58,7 +58,13 @@ def _get_points_image_space(self):
     def _get_points_body_space(self):
         template_matches, _ = self._get_template_matches()
 
-        return self._pdf.get_points_body_space(template_matches.cpu().numpy())
+        num_points, _ = template_matches.shape
+        objp = np.zeros((num_points, 3), np.float32)
+        objp[:, :2] = self._pdf.get_physical_measurements(
+            template_matches.cpu().numpy()
+        )
+
+        return objp
 
     def is_valid(self):
         template_matches, _ = self._get_template_matches()