add sam2

2024-08-12 10:10:24 +08:00
parent 9afdbd1c0a
commit 2f833029aa
23 changed files with 3801 additions and 3 deletions
--- a/iopaint/schema.py
+++ b/iopaint/schema.py
@@ -167,6 +167,10 @@ class InteractiveSegModel(Choices):
    sam_hq_vit_l = "sam_hq_vit_l"
    sam_hq_vit_h = "sam_hq_vit_h"
    mobile_sam = "mobile_sam"
+    sam2_tiny = "sam2_tiny"
+    sam2_small = "sam2_small"
+    sam2_base = "sam2_base"
+    sam2_large = "sam2_large"


 class PluginInfo(BaseModel):