HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Object-Aware Distillation Pyramid for Open-Vocabulary Object Detection

Luting Wang Yi Liu Penghui Du Zihan Ding Yue Liao Qiaosong Qi Biaolong Chen Si Liu

Object-Aware Distillation Pyramid for Open-Vocabulary Object Detection

Abstract

Open-vocabulary object detection aims to provide object detectors trained on a fixed set of object categories with the generalizability to detect objects described by arbitrary text queries. Previous methods adopt knowledge distillation to extract knowledge from Pretrained Vision-and-Language Models (PVLMs) and transfer it to detectors. However, due to the non-adaptive proposal cropping and single-level feature mimicking processes, they suffer from information destruction during knowledge extraction and inefficient knowledge transfer. To remedy these limitations, we propose an Object-Aware Distillation Pyramid (OADP) framework, including an Object-Aware Knowledge Extraction (OAKE) module and a Distillation Pyramid (DP) mechanism. When extracting object knowledge from PVLMs, the former adaptively transforms object proposals and adopts object-aware mask attention to obtain precise and complete knowledge of objects. The latter introduces global and block distillation for more comprehensive knowledge transfer to compensate for the missing relation information in object distillation. Extensive experiments show that our method achieves significant improvement compared to current methods. Especially on the MS-COCO dataset, our OADP framework reaches $35.6$ mAP$^{\text{N}}{50}$, surpassing the current state-of-the-art method by $3.3$ mAP$^{\text{N}}{50}$. Code is released at https://github.com/LutingWang/OADP.

Code Repositories

lutingwang/oadp
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
open-vocabulary-object-detection-on-lvis-v1-0OADP
AP novel-LVIS base training: 21.7
open-vocabulary-object-detection-on-mscocoOADP
AP 0.5: 30.0
open-vocabulary-object-detection-on-mscocoOADP (G-OVD)
AP 0.5: 35.6

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp