mirrors
/
segment-anything
mirror of https://github.com/facebookresearch/segment-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update return type hint of predict function

pull/188/head
derekray311511 2023-04-07 18:06:10 +08:00 committed by ericmintun
parent 40df6e4046
commit d4ecc68dea
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
segment_anything/predictor.py
View File

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