|
|
|
@ -97,7 +97,7 @@ class SamPredictor:
@@ -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. |
|
|
|
|
|
|
|
|
|