GRiT: A Generative Region-to-text Transformer for Object Understanding

December 01, 2022 · Entered Twilight · 🏛 European Conference on Computer Vision

Repo contents: .gitignore, .gitmodules, LICENSE, README.md, configs, datasets, demo.py, demo_images, docs, grit, lauch_deepspeed.py, requirements.txt, third_party, train_deepspeed.py, train_net.py

Authors Jialian Wu, Jianfeng Wang, Zhengyuan Yang, Zhe Gan, Zicheng Liu, Junsong Yuan, Lijuan Wang arXiv ID 2212.00280 Category cs.CV: Computer Vision Citations 147 Venue European Conference on Computer Vision Repository https://github.com/JialianW/GRiT ⭐ 340 Last Checked 1 month ago

Abstract

This paper presents a Generative RegIon-to-Text transformer, GRiT, for object understanding. The spirit of GRiT is to formulate object understanding as <region, text> pairs, where region locates objects and text describes objects. For example, the text in object detection denotes class names while that in dense captioning refers to descriptive sentences. Specifically, GRiT consists of a visual encoder to extract image features, a foreground object extractor to localize objects, and a text decoder to generate open-set object descriptions. With the same model architecture, GRiT can understand objects via not only simple nouns, but also rich descriptive sentences including object attributes or actions. Experimentally, we apply GRiT to object detection and dense captioning tasks. GRiT achieves 60.4 AP on COCO 2017 test-dev for object detection and 15.5 mAP on Visual Genome for dense captioning. Code is available at https://github.com/JialianW/GRiT