quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions 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
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