Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: output description for countgd #298

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions vision_agent/tools/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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.
Expand Down Expand Up @@ -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
Expand All @@ -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.
Expand Down
Loading