debug

inference.py

@@ -39,7 +39,9 @@ class DepthPredictor:
         return output
     
     def generate_pcl(self, image):
+        print(image.shape)
         depth = self.predict(image)
+        print(depth.shape)
         # Step 2: Create an RGBD image from the RGB and depth image
         depth_o3d = o3d.geometry.Image(depth)
         image_o3d = o3d.geometry.Image(np.array(image))
@@ -49,6 +51,7 @@ class DepthPredictor:
         # Step 4: Convert PointCloud data to a NumPy array
         points = np.asarray(pcd.points)
         colors = np.asarray(pcd.colors)
+        print(points.shape, colors.shape)
         return points, colors
     
     def generate_fig(self, image):
@@ -94,6 +97,7 @@ class DepthPredictor:
     def generate_obj_masks(self, image, n_samples, masks, cube_size):
         # Generate a point cloud
         point_cloud, color_array = self.generate_pcl(image)
+        print(point_cloud.shape)
         mesh = o3d.geometry.TriangleMesh()
         # Create cubes and add them to the mesh
         cs = [(255,0,0),(0,255,0),(0,0,255)]