Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

OV-DQUO: Open-Vocabulary DETR with Denoising Text Query Training and Open-World Unknown Objects Supervision

About

Open-vocabulary detection aims to detect objects from novel categories beyond the base categories on which the detector is trained. However, existing open-vocabulary detectors trained on base category data tend to assign higher confidence to trained categories and confuse novel categories with the background. To resolve this, we propose OV-DQUO, an \textbf{O}pen-\textbf{V}ocabulary DETR with \textbf{D}enoising text \textbf{Q}uery training and open-world \textbf{U}nknown \textbf{O}bjects supervision. Specifically, we introduce a wildcard matching method. This method enables the detector to learn from pairs of unknown objects recognized by the open-world detector and text embeddings with general semantics, mitigating the confidence bias between base and novel categories. Additionally, we propose a denoising text query training strategy. It synthesizes foreground and background query-box pairs from open-world unknown objects to train the detector through contrastive learning, enhancing its ability to distinguish novel objects from the background. We conducted extensive experiments on the challenging OV-COCO and OV-LVIS benchmarks, achieving new state-of-the-art results of 45.6 AP50 and 39.3 mAP on novel categories respectively, without the need for additional training data. Models and code are released at \url{https://github.com/xiaomoguhz/OV-DQUO}

Junjie Wang, Bin Chen, Bin Kang, Yulin Li, YiChi Chen, Weizhi Xian, Huifeng Chang, Yong Xu• 2024

Related benchmarks

TaskDatasetResultRank
Object DetectionLVIS v1.0 (val)--
529
Object DetectionOV-COCO
AP50 (Novel)45.3
130
Open-vocabulary object detectionOV-LVIS v1.0 (test)
APr39.5
50
Open-vocabulary object detectionOV-COCO
AP@50 (Novel)45.6
31
Object DetectionCOCO Open Vocabulary
AP Novel45.6
17
Open-vocabulary object detectionOV-COCO (val)
Novel-class mAP5039.2
11
Open-vocabulary object detectionOV-COCO-C (test)
mAP@0.5 (Gauss)9.7
11
Object DetectionCOCO Novel Base All 2017
AP Novel45.6
9
Showing 8 of 8 rows

Other info

Code

Follow for update