fix some parameter passing errors in EfficientSAM (#473)

IDEA-Research · Mar 16, 2024 · 3dd1a30 · 3dd1a30
1 parent 16614c5
commit 3dd1a30
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/EfficientSAM/grounded_efficient_sam.py b/EfficientSAM/grounded_efficient_sam.py
@@ -38,7 +38,7 @@
     image=image,
     classes=CLASSES,
     box_threshold=BOX_THRESHOLD,
-    text_threshold=BOX_THRESHOLD
+    text_threshold=TEXT_THRESHOLD
 )
 
 # annotate image with detections

diff --git a/EfficientSAM/grounded_light_hqsam.py b/EfficientSAM/grounded_light_hqsam.py
@@ -44,7 +44,7 @@
     image=image,
     classes=CLASSES,
     box_threshold=BOX_THRESHOLD,
-    text_threshold=BOX_THRESHOLD
+    text_threshold=TEXT_THRESHOLD
 )
 
 # annotate image with detections

diff --git a/EfficientSAM/grounded_repvit_sam.py b/EfficientSAM/grounded_repvit_sam.py
@@ -42,7 +42,7 @@
     image=image,
     classes=CLASSES,
     box_threshold=BOX_THRESHOLD,
-    text_threshold=BOX_THRESHOLD
+    text_threshold=TEXT_THRESHOLD
 )
 
 # annotate image with detections