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

Open Vocabulary Object Detection with Proposal Mining and Prediction Equalization

About

Open-vocabulary object detection (OVD) aims to scale up vocabulary size to detect objects of novel categories beyond the training vocabulary. Recent work resorts to the rich knowledge in pre-trained vision-language models. However, existing methods are ineffective in proposal-level vision-language alignment. Meanwhile, the models usually suffer from confidence bias toward base categories and perform worse on novel ones. To overcome the challenges, we present MEDet, a novel and effective OVD framework with proposal mining and prediction equalization. First, we design an online proposal mining to refine the inherited vision-semantic knowledge from coarse to fine, allowing for proposal-level detection-oriented feature alignment. Second, based on causal inference theory, we introduce a class-wise backdoor adjustment to reinforce the predictions on novel categories to improve the overall OVD performance. Extensive experiments on COCO and LVIS benchmarks verify the superiority of MEDet over the competing approaches in detecting objects of novel categories, e.g., 32.6% AP50 on COCO and 22.4% mask mAP on LVIS.

Peixian Chen, Kekai Sheng, Mengdan Zhang, Mingbao Lin, Yunhang Shen, Shaohui Lin, Bo Ren, Ke Li• 2022

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)--
2454
Instance SegmentationLVIS v1.0 (val)--
189
Instance SegmentationLVIS (val)--
46
Object DetectionMS-COCO 48/17 base/novel
GZSD All AP5048
21
Open-vocabulary object detectionCOCO OVD (Generalized (17 + 48))
Novel AP5032.6
14
Open-vocabulary object detectionLVIS
APr22.4
7
Object DetectionCOCO retained 15 categories
AP50 (Retain)18.6
4
Object DetectionCOCO OVD retained 15 categories
AP50 Retained18.6
4
Showing 8 of 8 rows

Other info

Code

Follow for update