Skip to content

Commit

Permalink
fixed return example
Browse files Browse the repository at this point in the history
  • Loading branch information
@hrnn
hrnn committed Dec 9, 2024
1 parent 12062e0 commit 1219fef
Showing 1 changed file with 26 additions and 22 deletions.
48 changes: 26 additions & 22 deletions vision_agent/tools/tools.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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),
},
],
...
]
"""

Expand Down Expand Up @@ -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),
},
],
...
]
"""

Expand Down

0 comments on commit 1219fef

Please sign in to comment.