From d4ecc68deac20da2f9ae9b101f9528875c06093f Mon Sep 17 00:00:00 2001 From: derekray311511 Date: Fri, 7 Apr 2023 18:06:10 +0800 Subject: [PATCH] Update return type hint of predict function --- segment_anything/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/segment_anything/predictor.py b/segment_anything/predictor.py index 91b2ed3..6661d2a 100644 --- 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.