fix

inference.py

@@ -73,7 +73,7 @@ class DepthPredictor:
     def generate_obj(self, image, n_samples, mask):
         # Step 1: Create a point cloud
         point_cloud, color_array = self.generate_pcl(image)
-        point_cloud, color_array = point_cloud[mask.ravel()], color_array[mask.ravel()]
+        point_cloud, color_array = point_cloud[mask.ravel()[:-1]], color_array[mask.ravel()[:-1]]
         # sample 1000 points
         idxs = np.random.choice(len(point_cloud), int(n_samples))
         point_cloud = point_cloud[idxs]