fix: use nanmean. (LASR-at-Home#143)

jws-1 · Mar 11, 2024 · c7cbcbd · c7cbcbd
1 parent 92e25e3
commit c7cbcbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/helpers/cv2_pcl/src/cv2_pcl/__init__.py b/common/helpers/cv2_pcl/src/cv2_pcl/__init__.py
@@ -48,7 +48,7 @@ def seg_to_centroid(pcl: PointCloud2, xyseg: np.ndarray) -> np.ndarray:
     # Extract points of interest
     xyz_points = [pcl_xyz[x][y] for x, y in indices]
 
-    return np.mean(xyz_points, axis=0)
+    return np.nanmean(xyz_points, axis=0)
 
 
 def bb_to_centroid(pcl: PointCloud2, x: int, y: int, w: int, h: int) -> np.ndarray: