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/web_config.py
+++ b/iopaint/web_config.py
@@ -48,7 +48,7 @@ default_configs = dict(
    output_dir=None,
    quality=95,
    enable_interactive_seg=False,
-    interactive_seg_model=InteractiveSegModel.vit_b,
+    interactive_seg_model=InteractiveSegModel.sam2_1_tiny,
    interactive_seg_device=Device.cpu,
    enable_remove_bg=False,
    remove_bg_model=RemoveBGModel.briaai_rmbg_1_4,