Update return type hint of predict function

2023-04-07 18:06:10 +08:00 · 2023-04-07 18:06:10 +08:00 · d4ecc68dea
parent 40df6e4046
commit d4ecc68dea
1 changed files with 1 additions and 1 deletions
--- a/segment_anything/predictor.py
+++ b/segment_anything/predictor.py
@ -97,7 +97,7 @@ class SamPredictor:
        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.