Merge branch 'facebookresearch:main' into main

2023-04-09 19:16:56 +08:00
parent 7408767146 efeab7296a
commit ea0f68a6c0
7 changed files with 87 additions and 51 deletions
--- a/segment_anything/build_sam.py
+++ b/segment_anything/build_sam.py
@@ -45,8 +45,8 @@ def build_sam_vit_b(checkpoint=None):


 sam_model_registry = {
-    "default": build_sam,
-    "vit_h": build_sam,
+    "default": build_sam_vit_h,
+    "vit_h": build_sam_vit_h,
    "vit_l": build_sam_vit_l,
    "vit_b": build_sam_vit_b,
 }