fix

inference.py

@@ -149,9 +149,9 @@ class DepthPredictor:
     def generate_obj_rgb(self, image, n_samples, cube_size, max_depth, min_depth):
         # Step 1: Create a point cloud
         depth = self.predict(image)
+        image = np.array(image)
         depth = map_image_range(depth, min_depth, max_depth)
         point_cloud, color_array = PCL_rgb(image, depth)
-        image = np.array(image)
         idxs = np.random.choice(len(point_cloud), int(n_samples))
         point_cloud = point_cloud[idxs]
         color_array = color_array[idxs]