Update Amodal3R/pipelines/Amodal3R_image_to_3d.py

Amodal3R/pipelines/Amodal3R_image_to_3d.py

@@ -128,9 +128,9 @@ class Amodal3RImageTo3DPipeline(Pipeline):
         output = Image.fromarray((output * 255).astype(np.uint8))
         return output
     
-    def preprocess_image_w_mask(self, input: Image.Image, mask: Image.Image, kernel_size=3) -> Image.Image:
-        image = np.array(input).astype(np.float32) / 255
-        mask_ori = np.array(mask).astype(np.float32)
+    def preprocess_image_w_mask(self, input, mask, kernel_size=3) -> Image.Image:
+        image = input.astype(np.float32) / 255
+        mask_ori = mask.astype(np.float32)
         mask = (mask_ori < 127).astype(np.uint8)
         if kernel_size > 0:
             kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (kernel_size, kernel_size))
@@ -396,8 +396,8 @@ class Amodal3RImageTo3DPipeline(Pipeline):
     @torch.no_grad()
     def run_multi_image(
         self,
-        images: List[Image.Image],
-        masks: List[Image.Image],
+        images: List[np.ndarray],
+        masks: List[np.ndarray],
         num_samples: int = 1,
         seed: int = 42,
         sparse_structure_sampler_params: dict = {},