fix

inference.py

@@ -40,13 +40,13 @@ def PCL(mask, depth):
     print(np.unique(rgb_mask))
     depth_o3d = o3d.geometry.Image(depth)
     image_o3d = o3d.geometry.Image(rgb_mask)
-    print(len(depth_o3d))
-    print(len(image_o3d))
+    #print(len(depth_o3d))
+    #print(len(image_o3d))
     rgbd_image = o3d.geometry.RGBDImage.create_from_color_and_depth(image_o3d, depth_o3d, convert_rgb_to_intensity=False)
     # Step 3: Create a PointCloud from the RGBD image
     pcd = o3d.geometry.PointCloud.create_from_rgbd_image(rgbd_image, o3d.camera.PinholeCameraIntrinsic(o3d.camera.PinholeCameraIntrinsicParameters.PrimeSenseDefault))
     # Step 4: Convert PointCloud data to a NumPy array
-    print(len(pcd))
+    #print(len(pcd))
     points = np.asarray(pcd.points)
     colors = np.asarray(pcd.colors)
     print(np.unique(colors, axis=0))