add MobileSAM vit_t

2023-10-07 12:48:29 +08:00
parent 94211a4985
commit 557e28aff9
4 changed files with 892 additions and 4 deletions
--- a/lama_cleaner/plugins/interactive_seg.py
+++ b/lama_cleaner/plugins/interactive_seg.py
@@ -22,6 +22,10 @@ SEGMENT_ANYTHING_MODELS = {
        "url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
        "md5": "4b8939a88964f0f4ff5f5b2642c598a6",
    },
+    "vit_t": {
+        "url": "https://github.com/Sanster/models/releases/download/MobileSAM/mobile_sam.pt",
+        "md5": "f3c0d8cda613564d499310dab6c812cd",
+    },
 }