From 1219fefd69e7b79766a511e8b64957cfae3ccd68 Mon Sep 17 00:00:00 2001 From: hrnn Date: Mon, 9 Dec 2024 16:27:25 -0400 Subject: [PATCH] fixed return example --- vision_agent/tools/tools.py | 48 ++++++++++++++++++++----------------- 1 file changed, 26 insertions(+), 22 deletions(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 36fafdb5..ef03c8a3 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -2738,18 +2738,20 @@ def countgd_sam2_video_tracking( Example ------- - >>> countgd_sam2_video_tracking("car, dinosaur", image) + >>> countgd_sam2_video_tracking("car, dinosaur", frames) [ - { - 'score': 1.0, - 'label': 'dinosaur', - 'bbox': [0.1, 0.11, 0.35, 0.4], - 'mask': array([[0, 0, 0, ..., 0, 0, 0], - [0, 0, 0, ..., 0, 0, 0], - ..., - [0, 0, 0, ..., 0, 0, 0], - [0, 0, 0, ..., 0, 0, 0]], dtype=uint8), - }, + [ + { + 'label': '0: dinosaur', + 'bbox': [0.1, 0.11, 0.35, 0.4], + 'mask': array([[0, 0, 0, ..., 0, 0, 0], + [0, 0, 0, ..., 0, 0, 0], + ..., + [0, 0, 0, ..., 0, 0, 0], + [0, 0, 0, ..., 0, 0, 0]], dtype=uint8), + }, + ], + ... ] """ @@ -2783,18 +2785,20 @@ def owlv2_sam2_video_tracking( Example ------- - >>> countgd_sam2_video_tracking("car, dinosaur", image) + >>> countgd_sam2_video_tracking("car, dinosaur", frames) [ - { - 'score': 1.0, - 'label': 'dinosaur', - 'bbox': [0.1, 0.11, 0.35, 0.4], - 'mask': array([[0, 0, 0, ..., 0, 0, 0], - [0, 0, 0, ..., 0, 0, 0], - ..., - [0, 0, 0, ..., 0, 0, 0], - [0, 0, 0, ..., 0, 0, 0]], dtype=uint8), - }, + [ + { + 'label': '0: dinosaur', + 'bbox': [0.1, 0.11, 0.35, 0.4], + 'mask': array([[0, 0, 0, ..., 0, 0, 0], + [0, 0, 0, ..., 0, 0, 0], + ..., + [0, 0, 0, ..., 0, 0, 0], + [0, 0, 0, ..., 0, 0, 0]], dtype=uint8), + }, + ], + ... ] """