arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1706.02430cs.CV

Image Captioning with Object Detection and Localization

Department of Electronic Engineering, Tsinghua University(清华大学电子工程系)

Zhongliang Yang, Yu-Jin Zhang, Sadaqat ur Rehman, Yongfeng Huang

英文摘要:

Automatically generating a natural language description of an image is a task close to the heart of image understanding. In this paper, we present a multi-model neural network method closely related to the human visual system that automatically learns to describe the content of images. Our model consists of two sub-models: an object detection and localization model, which extract the information of objects and their spatial relationship in images respectively; Besides, a deep recurrent neural network (RNN) based on long short-term memory (LSTM) units with attention mechanism for sentences generation. Each word of the description will be automatically aligned to different objects of the input image when it is generated. This is similar to the attention mechanism of the human visual system. Experimental results on the COCO dataset showcase the merit of the proposed method, which outperforms previous benchmark models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑