HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

RegionCLIP: Region-based Language-Image Pretraining

RegionCLIP: Region-based Language-Image Pretraining

Abstract

Contrastive language-image pretraining (CLIP) using image-text pairs has achieved impressive results on image classification in both zero-shot and transfer learning settings. However, we show that directly applying such models to recognize image regions for object detection leads to poor performance due to a domain shift: CLIP was trained to match an image as a whole to a text description, without capturing the fine-grained alignment between image regions and text spans. To mitigate this issue, we propose a new method called RegionCLIP that significantly extends CLIP to learn region-level visual representations, thus enabling fine-grained alignment between image regions and textual concepts. Our method leverages a CLIP model to match image regions with template captions and then pretrains our model to align these region-text pairs in the feature space. When transferring our pretrained model to the open-vocabulary object detection tasks, our method significantly outperforms the state of the art by 3.8 AP50 and 2.2 AP for novel categories on COCO and LVIS datasets, respectively. Moreoever, the learned region representations support zero-shot inference for object detection, showing promising results on both COCO and LVIS datasets. Our code is available at https://github.com/microsoft/RegionCLIP.

Code Repositories

microsoft/regionclip
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
open-vocabulary-object-detection-on-lvis-v1-0Region-CLIP (RN50-C4)
AP novel-LVIS base training: 17.1
open-vocabulary-object-detection-on-lvis-v1-0Region-CLIP (RN50x4-C4)
AP novel-LVIS base training: 22.0
open-vocabulary-object-detection-on-mscocoRegion-CLIP (RN50-C4)
AP 0.5: 31.4
open-vocabulary-object-detection-on-mscocoRegion-CLIP (RN50x4-C4)
AP 0.5: 39.3

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
RegionCLIP: Region-based Language-Image Pretraining | Papers | HyperAI