zjt/SAMapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update return type hint of predict function

Browse Source
This commit is contained in:
derekray311511
2023-04-12 09:50:19 -07:00
committed by ericmintun
parent 40df6e4046
commit d4ecc68dea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
segment_anything/predictor.py
+1 -1
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.