Browse Source

Update return type hint of predict function

pull/188/head
derekray311511 2 years ago committed by ericmintun
parent
commit
d4ecc68dea
  1. 2
      segment_anything/predictor.py

2
segment_anything/predictor.py

@ -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.

Loading…
Cancel
Save