feat: sam2.1

luca-medeiros · Oct 17, 2024 · 3be067c · 3be067c
1 parent 3347db9
commit 3be067c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 14 deletions.
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@ Language Segment-Anything is an open-source project that combines the power of i
 
 - Zero-shot text-to-bbox approach for object detection.
 - GroundingDINO detection model integration.
-- SAM 2.
+- SAM 2.1
 - Batch inference support.
 - Easy endpoint deployment using the Lightning AI litserve platform.
 - Customizable text prompts for precise object segmentation.

diff --git a/lang_sam/models/sam.py b/lang_sam/models/sam.py
@@ -18,21 +18,21 @@
 
 
 SAM_MODELS = {
-    "sam2_hiera_tiny": {
-        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt",
-        "config": "sam2_hiera_t.yaml",
+    "sam2.1_hiera_tiny": {
+        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt",
+        "config": "configs/sam2.1/sam2.1_hiera_t.yaml",
     },
-    "sam2_hiera_small": {
-        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_small.pt",
-        "config": "sam2_hiera_s.yaml",
+    "sam2.1_hiera_small": {
+        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt",
+        "config": "configs/sam2.1/sam2.1_hiera_s.yaml",
     },
-    "sam2_hiera_base_plus": {
-        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_base_plus.pt",
-        "config": "sam2_hiera_b+.yaml",
+    "sam2.1_hiera_base_plus": {
+        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt",
+        "config": "configs/sam2.1/sam2.1_hiera_b+.yaml",
     },
-    "sam2_hiera_large": {
-        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_large.pt",
-        "config": "sam2_hiera_l.yaml",
+    "sam2.1_hiera_large": {
+        "url": "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt",
+        "config": "configs/sam2.1/sam2.1_hiera_l.yaml",
     },
 }
 

diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ gradio==5.0.2
 litserve==0.2.3
 opencv-python-headless==4.10.0.84
 pydantic==2.9.2
-sam-2 @ git+https://github.com/facebookresearch/segment-anything-2@7e1596c0b6462eb1d1ba7e1492430fed95023598
+sam-2 @ git+https://github.com/facebookresearch/segment-anything-2@c2ec8e14a185632b0a5d8b161928ceb50197eddc
 supervision==0.23.0 ; python_full_version > '3.10'
 transformers==4.44.2
 uvloop==0.20.0