Skip to content

Commit

Permalink
fixed ordering of args (#200)
Browse files Browse the repository at this point in the history
  • Loading branch information
dillonalaird authored Aug 15, 2024
1 parent 0cf3037 commit e047bc2
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions vision_agent/tools/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -607,12 +607,13 @@ def florencev2_image_caption(image: np.ndarray, detail_caption: bool = True) ->
return answer[task] # type: ignore


def florencev2_object_detection(image: np.ndarray, prompt: str) -> List[Dict[str, Any]]:
def florencev2_object_detection(prompt: str, image: np.ndarray) -> List[Dict[str, Any]]:
"""'florencev2_object_detection' is a tool that can detect objects given a text
prompt such as a phrase or class names separated by commas. It returns a list of
detected objects as labels and their location as bounding boxes with score of 1.0.
Parameters:
prompt (str): The prompt to ground to the image.
image (np.ndarray): The image to used to detect objects
Returns:
Expand All @@ -624,7 +625,7 @@ def florencev2_object_detection(image: np.ndarray, prompt: str) -> List[Dict[str
Example
-------
>>> florencev2_object_detection(image, 'person looking at a coyote')
>>> florencev2_object_detection('person looking at a coyote', image)
[
{'score': 1.0, 'label': 'person', 'bbox': [0.1, 0.11, 0.35, 0.4]},
{'score': 1.0, 'label': 'coyote', 'bbox': [0.34, 0.21, 0.85, 0.5},
Expand Down

0 comments on commit e047bc2

Please sign in to comment.