From e9c47dd235ad5e558340e5457b2fd39d97d023fb Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 7 May 2024 20:14:51 -0700 Subject: [PATCH] renamed viz tools --- vision_agent/tools/tools_v2.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/vision_agent/tools/tools_v2.py b/vision_agent/tools/tools_v2.py index 2016410e..0754dea1 100644 --- a/vision_agent/tools/tools_v2.py +++ b/vision_agent/tools/tools_v2.py @@ -272,7 +272,7 @@ def save_image(image: np.ndarray) -> str: return f.name -def display_bounding_boxes( +def overlay_bounding_boxes( image: np.ndarray, bboxes: List[Dict[str, Any]] ) -> np.ndarray: """'display_bounding_boxes' is a utility function that displays bounding boxes on @@ -326,7 +326,7 @@ def display_bounding_boxes( return np.array(pil_image.convert("RGB")) -def display_segmentation_masks( +def overlay_segmentation_masks( image: np.ndarray, masks: List[Dict[str, Any]] ) -> np.ndarray: """'display_segmentation_masks' is a utility function that displays segmentation @@ -418,12 +418,12 @@ def get_tools_df(funcs: List[Callable[..., Any]]) -> pd.DataFrame: ocr, load_image, save_image, - display_bounding_boxes, - display_segmentation_masks, + overlay_bounding_boxes, + overlay_segmentation_masks, ] TOOLS_DF = get_tools_df(TOOLS) # type: ignore TOOL_DESCRIPTIONS = get_tool_descriptions(TOOLS) # type: ignore TOOL_DOCSTRING = get_tool_documentation(TOOLS) # type: ignore UTILITIES_DOCSTRING = get_tool_documentation( - [load_image, save_image, display_bounding_boxes] + [load_image, save_image, overlay_bounding_boxes] )