T-Rex2: Towards Generic Object Detection via Text-Visual Prompt Synergy
This work addresses the problem of detecting rare or complex objects in open-set scenarios for computer vision applications, representing an incremental improvement by combining existing prompt modalities.
The paper tackles the challenge of open-set object detection by introducing T-Rex2, a model that synergizes text and visual prompts through contrastive learning to handle diverse scenarios, achieving remarkable zero-shot detection capabilities across a wide spectrum of real-world applications.
We present T-Rex2, a highly practical model for open-set object detection. Previous open-set object detection methods relying on text prompts effectively encapsulate the abstract concept of common objects, but struggle with rare or complex object representation due to data scarcity and descriptive limitations. Conversely, visual prompts excel in depicting novel objects through concrete visual examples, but fall short in conveying the abstract concept of objects as effectively as text prompts. Recognizing the complementary strengths and weaknesses of both text and visual prompts, we introduce T-Rex2 that synergizes both prompts within a single model through contrastive learning. T-Rex2 accepts inputs in diverse formats, including text prompts, visual prompts, and the combination of both, so that it can handle different scenarios by switching between the two prompt modalities. Comprehensive experiments demonstrate that T-Rex2 exhibits remarkable zero-shot object detection capabilities across a wide spectrum of scenarios. We show that text prompts and visual prompts can benefit from each other within the synergy, which is essential to cover massive and complicated real-world scenarios and pave the way towards generic object detection. Model API is now available at \url{https://github.com/IDEA-Research/T-Rex}.