423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-09-18 04:32:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update docstring

Browse Source
This commit is contained in:
Niels 2024-08-05 09:44:03 +02:00
parent e93be7f6aa
commit 841cc1f015
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sam2/sam2_video_predictor.py
View File

@ -104,7 +104,7 @@ class SAM2VideoPredictor(SAM2Base):
return inference_state
@classmethod
def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2ImagePredictor":
def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2VideoPredictor":
"""
Load a pretrained model from the Hugging Face model hub.
@ -113,7 +113,7 @@ class SAM2VideoPredictor(SAM2Base):
**kwargs: Additional arguments to pass to the model constructor.
Returns:
(SAM2ImagePredictor): The loaded model.
(SAM2VideoPredictor): The loaded model.
"""
from sam2.build_sam import build_sam2_video_predictor_hf