value_map cone generation fixed; treating depths as distances along o…

…ptical axis, not euclidean distances
bdaiinstitute · Aug 18, 2023 · 541c742 · 541c742
1 parent d638da8
commit 541c742
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zsos/mapping/value_map.py b/zsos/mapping/value_map.py
@@ -230,8 +230,8 @@ def _get_visible_mask(self, depth: np.ndarray) -> np.ndarray:
 
         # Assign each value in the row with an x, y coordinate depending on 'angles'
         # and the max depth value for that column
-        x = depth_row * np.cos(angles)
-        y = depth_row * np.sin(angles)
+        x = depth_row
+        y = depth_row * np.tan(angles)
 
         # Get blank cone mask
         cone_mask = self._get_confidence_mask()