boolean dimension fix

inference.py

@@ -98,7 +98,11 @@ class DepthPredictor:
         # Create cubes and add them to the mesh
         cs = [(255,0,0),(0,255,0),(0,0,255)]
         for c,(mask, _) in zip(cs, masks):
-            mask = mask.ravel()[:-1]
+            if len(mask) == len(point_cloud):
+                mask = mask.ravel()
+            else:
+                mask = mask.ravel()[:-1]
+                
             point_cloud_subset, color_array_subset = point_cloud[mask], color_array[mask]
             idxs = np.random.choice(len(point_cloud_subset), int(n_samples))
             point_cloud_subset = point_cloud_subset[idxs]