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

CORA: Adapting CLIP for Open-Vocabulary Detection with Region Prompting and Anchor Pre-Matching

About

Open-vocabulary detection (OVD) is an object detection task aiming at detecting objects from novel categories beyond the base categories on which the detector is trained. Recent OVD methods rely on large-scale visual-language pre-trained models, such as CLIP, for recognizing novel objects. We identify the two core obstacles that need to be tackled when incorporating these models into detector training: (1) the distribution mismatch that happens when applying a VL-model trained on whole images to region recognition tasks; (2) the difficulty of localizing objects of unseen classes. To overcome these obstacles, we propose CORA, a DETR-style framework that adapts CLIP for Open-vocabulary detection by Region prompting and Anchor pre-matching. Region prompting mitigates the whole-to-region distribution gap by prompting the region features of the CLIP-based region classifier. Anchor pre-matching helps learning generalizable object localization by a class-aware matching mechanism. We evaluate CORA on the COCO OVD benchmark, where we achieve 41.7 AP50 on novel classes, which outperforms the previous SOTA by 2.4 AP50 even without resorting to extra training data. When extra training data is available, we train CORA$^+$ on both ground-truth base-category annotations and additional pseudo bounding box labels computed by CORA. CORA$^+$ achieves 43.1 AP50 on the COCO OVD benchmark and 28.1 box APr on the LVIS OVD benchmark.

Xiaoshi Wu, Feng Zhu, Rui Zhao, Hongsheng Li• 2023

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)--
2454
Object DetectionLVIS v1.0 (val)--
518
Instance SegmentationLVIS v1.0 (val)--
189
Object DetectionOV-COCO
AP50 (Novel)43.1
97
Instance SegmentationLVIS--
68
Instance SegmentationLVIS (val)
APr28.1
46
Object DetectionFSOD
mAP5049.3
39
Object DetectioniNatLoc
mAP5031.2
36
Object DetectionOV-LVIS v1.0 (test)
mAPr28.1
27
Dynamic Object DetectionD³ (Full)
Intra-scenario mAP6.2
20
Showing 10 of 25 rows

Other info

Code

Follow for update