RTGen: Generating Region-Text Pairs for Open-Vocabulary Object Detection

Fangyi Chen, Han Zhang, Zhantao Yang, Hao Chen, Kai Hu +1 more
5/30/2024
cs.CV

No official code found. Best implementation: seermer/RTGen (14 stars, last commit 2024-07-30).

Abstract

Open-vocabulary object detection (OVD) requires solid modeling of the region-semantic relationship, which could be learned from massive region-text pairs. However, such data is limited in practice due to significant annotation costs. In this work, we propose RTGen to generate scalable open-vocabulary region-text pairs and demonstrate its capability to boost the performance of open-vocabulary object detection. RTGen includes both text-to-region and region-to-text generation processes on scalable image-caption data. The text-to-region generation is powered by image inpainting, directed by our proposed scene-aware inpainting guider for overall layout harmony. For region-to-text generation, we perform multiple region-level image captioning with various prompts and select the best matching text according to CLIP similarity. To facilitate detection training on region-text pairs, we also introduce a localization-aware region-text contrastive loss that learns object proposals tailored with different localization qualities. Extensive experiments demonstrate that our RTGen can serve as a scalable, semantically rich, and effective source for open-vocabulary object detection and continue to improve the model performance when more data is utilized, delivering superior performance compared to the existing state-of-the-art methods.

View on arXivView PDF

Code Implementations(1)

seermer/RTGenCommunity100%
140Shell, PythonMay 22, 20242 years ago

Similar Papers

Cite this paper

@article{chen2024rtgen,
  title  = {RTGen: Generating Region-Text Pairs for Open-Vocabulary Object Detection},
  author = {Fangyi Chen and Han Zhang and Zhantao Yang and Hao Chen and Kai Hu and Marios Savvides},
  year   = {2024},
  eprint = {2405.19854},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.19854v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.19854.svg)](https://codeofpaper.com/paper/2405.19854)

Discussion