SAM 2.1

Author: cpuhrsch

examples/sam2_amg_server/sam2_hiera_l.yaml

@@ -60,13 +60,12 @@ def main(checkpoint_path):
 
     device = "cuda"
     from pathlib import Path
-    sam2_checkpoint = Path(checkpoint_path) / Path("sam2_hiera_large.pt")
-    model_cfg = "sam2_hiera_l.yaml"
+    sam2_checkpoint = Path(checkpoint_path) / Path("sam2.1_hiera_large.pt")
+    model_cfg = "configs/sam2.1/sam2.1_hiera_l.yaml"
     logging.basicConfig(level=logging.INFO)
 
-    logging.info(f"Loading model: {sam2_checkpoint}")
+    logging.info(f"Loading model {sam2_checkpoint} with config {model_cfg}")
     sam2 = build_sam2(model_cfg, sam2_checkpoint, device=device, apply_postprocessing=False)
-    sam2.to(device=device)
 
     mask_generator = SAM2AutomaticMaskGenerator(sam2) #, points_per_batch=None)