add sam_hq

2024-02-20 09:03:11 +08:00
parent 6447e821cb
commit b358e6cbce
10 changed files with 1281 additions and 19 deletions
--- a/iopaint/plugins/segment_anything/init.py
+++ b/iopaint/plugins/segment_anything/init.py
@@ -5,10 +5,12 @@
 # LICENSE file in the root directory of this source tree.

 from .build_sam import (
-    build_sam,
    build_sam_vit_h,
    build_sam_vit_l,
    build_sam_vit_b,
+    build_sam_vit_h_hq,
+    build_sam_vit_l_hq,
+    build_sam_vit_b_hq,
    sam_model_registry,
 )
 from .predictor import SamPredictor