423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-11-04 11:32:12 +08:00
Code Issues 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
Show Stats 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>)