Download onnx models from Github release of efficient-sam

wkentaro · Dec 27, 2023 · 5b692a7 · 5b692a7
1 parent 05812d5
commit 5b692a7
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/labelme/ai/__init__.py b/labelme/ai/__init__.py
@@ -57,8 +57,14 @@ class EfficientSamVitT(EfficientSam):
 
     def __init__(self):
         super().__init__(
-            encoder_path="/Users/wkentaro/Projects/Labelme/efficient-sam/weights/efficient_sam_vitt_encoder.onnx",
-            decoder_path="/Users/wkentaro/Projects/Labelme/efficient-sam/weights/efficient_sam_vitt_decoder.onnx",
+            encoder_path=gdown.cached_download(
+                url="https://github.com/labelmeai/efficient-sam/releases/download/onnx-models-20231225/efficient_sam_vitt_encoder.onnx",  # NOQA
+                md5="2d4a1303ff0e19fe4a8b8ede69c2f5c7",
+            ),
+            decoder_path=gdown.cached_download(
+                url="https://github.com/labelmeai/efficient-sam/releases/download/onnx-models-20231225/efficient_sam_vitt_decoder.onnx",  # NOQA
+                md5="be3575ca4ed9b35821ac30991ab01843",
+            ),
         )
 
 
@@ -67,8 +73,14 @@ class EfficientSamVitS(EfficientSam):
 
     def __init__(self):
         super().__init__(
-            encoder_path="/Users/wkentaro/Projects/Labelme/efficient-sam/weights/efficient_sam_vits_encoder.onnx",
-            decoder_path="/Users/wkentaro/Projects/Labelme/efficient-sam/weights/efficient_sam_vits_decoder.onnx",
+            encoder_path=gdown.cached_download(
+                url="https://github.com/labelmeai/efficient-sam/releases/download/onnx-models-20231225/efficient_sam_vits_encoder.onnx",  # NOQA
+                md5="7d97d23e8e0847d4475ca7c9f80da96d",
+            ),
+            decoder_path=gdown.cached_download(
+                url="https://github.com/labelmeai/efficient-sam/releases/download/onnx-models-20231225/efficient_sam_vits_decoder.onnx",  # NOQA
+                md5="d9372f4a7bbb1a01d236b0508300b994",
+            ),
         )