facebook
/

sam2.1-hiera-large

Mask Generation

feature-extraction

Model card Files Files and versions

yonigozlan HF Staff commited on Aug 14

Commit

756baa3

·

1 Parent(s): 7ca4132

sam2 -> sam2.1

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ For image prediction:
 import torch
 from sam2.sam2_image_predictor import SAM2ImagePredictor
-predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     predictor.set_image(<your_image>)
@@ -29,7 +29,7 @@ For video prediction:
 import torch
 from sam2.sam2_video_predictor import SAM2VideoPredictor
-predictor = SAM2VideoPredictor.from_pretrained("facebook/sam2-hiera-large")
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     state = predictor.init_state(<your_video>)

 import torch
 from sam2.sam2_image_predictor import SAM2ImagePredictor
+predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large")
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     predictor.set_image(<your_image>)
 import torch
 from sam2.sam2_video_predictor import SAM2VideoPredictor
+predictor = SAM2VideoPredictor.from_pretrained("facebook/sam2.1-hiera-large")
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     state = predictor.init_state(<your_video>)