debug mode

inference.py

@@ -19,6 +19,7 @@ def PCL(mask, depth):
     assert type(depth) == np.ndarray
     rgb_mask = np.zeros((mask.shape[0], mask.shape[1], 3)).astype("uint8")
     rgb_mask[mask] = (255, 0, 0)
+    print
     depth_o3d = o3d.geometry.Image(depth)
     image_o3d = o3d.geometry.Image(rgb_mask)
     rgbd_image = o3d.geometry.RGBDImage.create_from_color_and_depth(image_o3d, depth_o3d, convert_rgb_to_intensity=False)
@@ -27,7 +28,12 @@ def PCL(mask, depth):
     # Step 4: Convert PointCloud data to a NumPy array
     points = np.asarray(pcd.points)
     colors = np.asarray(pcd.colors)
+    print(np.unique(colors, axis=0))
+    print(np.unique(colors, axis=1))
+    print(np.unique(colors))
     mask = (colors[:, 0] == 255)
+    print(mask.sum())
+    print(colors.shape)
     points = points[mask]
     colors = colors[mask]
     return points, colors