Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Simple Open-Vocabulary Object Detection with Vision Transformers

About

Combining simple architectures with large-scale pre-training has led to massive improvements in image classification. For object detection, pre-training and scaling approaches are less well established, especially in the long-tailed and open-vocabulary setting, where training data is relatively scarce. In this paper, we propose a strong recipe for transferring image-text models to open-vocabulary object detection. We use a standard Vision Transformer architecture with minimal modifications, contrastive image-text pre-training, and end-to-end detection fine-tuning. Our analysis of the scaling properties of this setup shows that increasing image-level pre-training and model size yield consistent improvements on the downstream detection task. We provide the adaptation strategies and regularizations needed to attain very strong performance on zero-shot text-conditioned and one-shot image-conditioned object detection. Code and models are available on GitHub.

Matthias Minderer, Alexey Gritsenko, Austin Stone, Maxim Neumann, Dirk Weissenborn, Alexey Dosovitskiy, Aravindh Mahendran, Anurag Arnab, Mostafa Dehghani, Zhuoran Shen, Xiao Wang, Xiaohua Zhai, Thomas Kipf, Neil Houlsby• 2022

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP43.5
2454
Object DetectionLVIS v1.0 (val)
APbbox35.3
518
Referring Expression ComprehensionRefCOCO (testA)--
333
Referring Expression ComprehensionRefCOCO+ (testA)--
207
Object DetectionLVIS (minival)
AP34.6
127
Object DetectionODinW-13
AP40.9
98
Object DetectionOV-COCO
AP50 (Novel)41.8
97
Instance SegmentationLVIS
mAP (Mask)34.7
68
Object DetectionLVIS
APr31.2
59
Object DetectionODinW-35
AP18.8
59
Showing 10 of 32 rows

Other info

Code

Follow for update