checkpoint wrong url

Jordan-Pierce · Nov 7, 2024 · ed0a6db · ed0a6db
1 parent 79674ce
commit ed0a6db
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -80,7 +80,7 @@ download_asset(model_url, model_path)
 ### Model Checkpoint URLs:
 - [edge_sam](https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam.pth)
 - [edge_sam_3x](https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam_3x.pth)
-- [vit_t](https://huggingface.co/spaces/dhkim2810/MobileSAM/blob/main/mobile_sam.pt)
+- [vit_t](https://huggingface.co/spaces/dhkim2810/MobileSAM/resolve/main/mobile_sam.pt)
 - [vit_b](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth)
 - [vit_l](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth)
 - [vit_h](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)

diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name="x_segment_anything",
-    version="0.0.3",
+    version="0.0.4",
     author="Jordan Pierce",
     author_email="jordan.pierce@noaa.gov",
     url="https://github.com/Jordan-Pierce/xSAM",

diff --git a/x_segment_anything/build_sam.py b/x_segment_anything/build_sam.py
@@ -26,6 +26,9 @@ def build_sam_vit_h(checkpoint=None):
     return _build_sam(image_encoder, checkpoint)
 
 
+build_sam = build_sam_vit_h
+
+
 def build_sam_vit_l(checkpoint=None):
     image_encoder = _build_sam_encoder(
         encoder_embed_dim=1024,
@@ -113,8 +116,8 @@ def build_edge_sam(checkpoint=None, upsample_mode="bicubic"):
     "vit_h": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
     "vit_l": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth",
     "vit_b": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth",
-    "vit_t": "https://huggingface.co/spaces/dhkim2810/MobileSAM/blob/main/mobile_sam.pt",
-    "edge_sam_3x": "https://huggingface.co/spaces/dhkim2810/MobileSAM/blob/main/mobile_sam.pt",
+    "vit_t": "https://huggingface.co/spaces/dhkim2810/MobileSAM/resolve/main/mobile_sam.pt",
+    "edge_sam_3x": "https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam_3x.pth",
     "edge_sam": "https://huggingface.co/spaces/chongzhou/EdgeSAM/resolve/main/weights/edge_sam.pth"
 }