Merge pull request #63 from Jize-W/add-bench-tools

register CelebrityRecognition tool
InternLM · Feb 19, 2024 · 70988df · 70988df
2 parents 0f84746 + d565726
commit 70988df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agentlego/tools/__init__.py b/agentlego/tools/__init__.py
@@ -8,7 +8,7 @@
 from .image_scribble import ImageToScribble, ScribbleTextToImage
 from .image_text import ImageDescription, TextToImage, ImageRegionDescription
 from .imagebind import AudioImageToImage, AudioTextToImage, AudioToImage, ThermalToImage
-from .object_detection import ObjectDetection, TextToBbox, CountGivenObject
+from .object_detection import ObjectDetection, TextToBbox, CountGivenObject, CelebrityRecognition
 from .ocr import OCR, MathOCR
 from .search import GoogleSearch
 from .segmentation import SegmentAnything, SegmentObject, SemanticSegmentation
@@ -25,5 +25,5 @@
     'SegmentObject', 'SegmentAnything', 'SemanticSegmentation', 'ImageStylization',
     'AudioToImage', 'ThermalToImage', 'AudioImageToImage', 'AudioTextToImage',
     'SpeechToText', 'TextToSpeech', 'Translation', 'GoogleSearch', 'Calculator',
-    'BaseTool', 'make_tool', 'AddText', 'DrawBox', 'ImageRegionDescription', 'CountGivenObject', 'ReplaceBackgroundOrForeground', 'MathOCR', 'PythonInterpreter', 'Plot', 'Solver'
+    'BaseTool', 'make_tool', 'AddText', 'DrawBox', 'ImageRegionDescription', 'CountGivenObject', 'ReplaceBackgroundOrForeground', 'MathOCR', 'PythonInterpreter', 'Plot', 'Solver', 'CelebrityRecognition'
 ]
diff --git a/agentlego/tools/object_detection/__init__.py b/agentlego/tools/object_detection/__init__.py
@@ -2,4 +2,4 @@
 from .text_to_bbox import TextToBbox
 from .count_given_object import CountGivenObject
 
-__all__ = ['ObjectDetection', 'TextToBbox', 'CountGivenObject']
+__all__ = ['ObjectDetection', 'TextToBbox', 'CountGivenObject', 'CelebrityRecognition']