quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Revert code snippet

Browse Source
This commit is contained in:
Niels
2024-08-06 22:57:07 +02:00
parent 8f15c6255a
commit 322aa3e7e5
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@@ -82,9 +82,11 @@ For promptable segmentation and tracking in videos, we provide a video predictor
```python
import torch
from sam2.sam2_video_predictor import SAM2VideoPredictor
from sam2.build_sam import build_sam2_video_predictor
predictor = SAM2VideoPredictor.from_pretrained("facebook/sam2-hiera-large")
checkpoint = "./checkpoints/sam2_hiera_large.pt"
model_cfg = "sam2_hiera_l.yaml"
predictor = build_sam2_video_predictor(model_cfg, checkpoint)
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
state = predictor.init_state(<your_video>)