Update return type hint of predict function

phygitalism · Apr 12, 2023 · d4ecc68 · d4ecc68
1 parent 40df6e4
commit d4ecc68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/segment_anything/predictor.py b/segment_anything/predictor.py
@@ -97,7 +97,7 @@ def predict(
         mask_input: Optional[np.ndarray] = None,
         multimask_output: bool = True,
         return_logits: bool = False,
-    ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
+    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
         """
         Predict masks for the given input prompts, using the currently set image.