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

Update README

Browse Source
This commit is contained in:
Niels
2024-08-05 09:43:04 +02:00
parent cb48213066
commit e93be7f6aa
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@@ -60,12 +60,9 @@ SAM 2 has all the capabilities of [SAM](https://github.com/facebookresearch/segm
```python ```python
import torch import torch
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor from sam2.sam2_image_predictor import SAM2ImagePredictor
checkpoint = "./checkpoints/sam2_hiera_large.pt" predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
model_cfg = "sam2_hiera_l.yaml"
predictor = SAM2ImagePredictor(build_sam2(model_cfg, checkpoint))
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16): with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
predictor.set_image(<your_image>) predictor.set_image(<your_image>)
@@ -82,11 +79,9 @@ For promptable segmentation and tracking in videos, we provide a video predictor
```python ```python
import torch import torch
from sam2.build_sam import build_sam2_video_predictor from sam2.sam2_video_predictor import SAM2VideoPredictor
checkpoint = "./checkpoints/sam2_hiera_large.pt" predictor = SAM2VideoPredictor.from_pretrained("facebook/sam2-hiera-large")
model_cfg = "sam2_hiera_l.yaml"
predictor = build_sam2_video_predictor(model_cfg, checkpoint)
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16): with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
state = predictor.init_state(<your_video>) state = predictor.init_state(<your_video>)