support box threshold in GD 1.5 demos

2024-10-31 15:50:14 +08:00
parent e537a1e763
commit 1aec7ded16
4 changed files with 7 additions and 1 deletions
--- a/grounded_sam2_tracking_demo_with_gd1.5.py
+++ b/grounded_sam2_tracking_demo_with_gd1.5.py
@@ -80,6 +80,7 @@ task = DetectionTask(
    prompts=[TextPrompt(text="children. pillow")],
    targets=[DetectionTarget.BBox],  # detect bbox
    model=DetectionModel.GDino1_5_Pro,  # detect with GroundingDino-1.5-Pro model
+    bbox_threshold=0.2,
 )

 client.run_task(task)