fix

inference.py

@@ -170,8 +170,8 @@ class DepthPredictor:
         cs = [(255,0,0),(0,255,0),(0,0,255)]
         for c,(mask, _) in zip(cs, masks):
             points, _ = PCL(mask, depth)
-            idxs = np.random.choice(len(points), int(n_samples))
-            points = points[idxs]
+            #idxs = np.random.choice(len(points), int(n_samples))
+            #points = points[idxs]
             for point in points:
                 cube = o3d.geometry.TriangleMesh.create_box(width=cube_size, height=cube_size, depth=cube_size)
                 cube.translate(-point)