Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 10, 2023

Commit

38b2f3b

1 Parent(s): c1b1bf8

fix

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -74,7 +74,7 @@ class DepthPredictor:
         # Step 1: Create a point cloud
         point_cloud, color_array = self.generate_pcl(image)
         # sample 1000 points
-        idxs = np.random.choice(len(point_cloud), 10e6)
         point_cloud = point_cloud[idxs]
         color_array = color_array[idxs]
         # Create a mesh to hold the colored cubes

         # Step 1: Create a point cloud
         point_cloud, color_array = self.generate_pcl(image)
         # sample 1000 points
+        idxs = np.random.choice(len(point_cloud), int(1e6))
         point_cloud = point_cloud[idxs]
         color_array = color_array[idxs]
         # Create a mesh to hold the colored cubes