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()[:-1]], color_array[mask.ravel()[:-1]]
+        #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]
@@ -83,7 +83,7 @@ class DepthPredictor:
         # Create cubes and add them to the mesh
         for point, color in zip(point_cloud, color_array):
             cube = o3d.geometry.TriangleMesh.create_box(width=0.000005, height=0.000005, depth=0.000005)
-            cube.translate(point)
+            cube.translate(-point)
             cube.paint_uniform_color(color)
             mesh += cube
         # Save the mesh to an .obj file