Merge pull request #128 from zivid/fix-normals-sample

Fix display_pointcloud function
zivid · Jan 4, 2022 · 2cdde1d · 2cdde1d
2 parents cfecc94 + 90e1554
commit 2cdde1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/sample_utils/display.py b/source/sample_utils/display.py
@@ -57,7 +57,8 @@ def display_pointcloud(xyz, rgb, normals=None):
 
     """
     xyz = np.nan_to_num(xyz).reshape(-1, 3)
-    normals = np.nan_to_num(normals).reshape(-1, 3)
+    if normals is not None:
+        normals = np.nan_to_num(normals).reshape(-1, 3)
     rgb = rgb.reshape(-1, 3)
 
     point_cloud_open3d = o3d.geometry.PointCloud(o3d.utility.Vector3dVector(xyz))