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

Update return type hint of predict function

parent 40df6e40
......@@ -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.
......
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