From 04604c0f81d22865491efb42652f58ca37f102dd Mon Sep 17 00:00:00 2001 From: hrnn Date: Mon, 11 Nov 2024 13:42:46 -0800 Subject: [PATCH] fix: output description for countgd --- vision_agent/tools/tools.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index cc8499a1..c3adb1ee 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -681,7 +681,7 @@ def countgd_counting( box_threshold: float = 0.23, ) -> List[Dict[str, Any]]: """'countgd_counting' is a tool that can precisely count multiple instances of an - object given a text prompt. It returns a list of bounding boxes with normalized + object given a text prompt. It returns a list of bounding boxes with unnormalized coordinates, label names and associated confidence scores. Parameters: @@ -692,7 +692,7 @@ def countgd_counting( Returns: List[Dict[str, Any]]: A list of dictionaries containing the score, label, and - bounding box of the detected objects with normalized coordinates between 0 + bounding box of the detected objects with unnormalized coordinates between 0 and 1 (xmin, ymin, xmax, ymax). xmin and ymin are the coordinates of the top-left and xmax and ymax are the coordinates of the bottom-right of the bounding box. @@ -746,7 +746,7 @@ def countgd_example_based_counting( ) -> List[Dict[str, Any]]: """'countgd_example_based_counting' is a tool that can precisely count multiple instances of an object given few visual example prompts. It returns a list of bounding - boxes with normalized coordinates, label names and associated confidence scores. + boxes with unnormalized coordinates, label names and associated confidence scores. Parameters: visual_prompts (List[List[float]]): Bounding boxes of the object in format @@ -757,7 +757,7 @@ def countgd_example_based_counting( Returns: List[Dict[str, Any]]: A list of dictionaries containing the score, label, and - bounding box of the detected objects with normalized coordinates between 0 + bounding box of the detected objects with unnormalized coordinates between 0 and 1 (xmin, ymin, xmax, ymax). xmin and ymin are the coordinates of the top-left and xmax and ymax are the coordinates of the bottom-right of the bounding box.