add sam2.1

2024-11-01 18:45:32 +08:00
parent 2098d687d1
commit ec08ce3063
11 changed files with 565 additions and 184 deletions
--- a/iopaint/schema.py
+++ b/iopaint/schema.py
@@ -175,11 +175,17 @@ 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"

+    sam2_1_tiny = "sam2_1_tiny"
+    sam2_1_small = "sam2_1_small"
+    sam2_1_base = "sam2_1_base"
+    sam2_1_large = "sam2_1_large"
+

 class PluginInfo(BaseModel):
    name: str