Skip to content

GitLab

Commit d4ecc68d authored by derekray311511's avatar derekray311511 Committed by ericmintun
Browse files

Update return type hint of predict function

parent 40df6e40
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
segment_anything/predictor.py
View file @ d4ecc68d
...@@ -97,7 +97,7 @@ class SamPredictor: ...@@ -97,7 +97,7 @@ class SamPredictor:
mask_input: Optional[np.ndarray] = None, mask_input: Optional[np.ndarray] = None,
multimask_output: bool = True, multimask_output: bool = True,
return_logits: bool = False, 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. Predict masks for the given input prompts, using the currently set image.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment