Enhanced image captioning with color recognition using deep learning methods

Yeong Hwa Chang*, Yen Jen Chen, Ren Hung Huang, Yi Ting Yu

*Corresponding author for this work

Research output: Contribution to journalJournal Article peer-review

12 Scopus citations

Abstract

Automatically describing the content of an image is an interesting and challenging task in artificial intelligence. In this paper, an enhanced image captioning model—including object detection, color analysis, and image captioning—is proposed to automatically generate the textual descriptions of images. In an encoder–decoder model for image captioning, VGG16 is used as an encoder and an LSTM (long short-term memory) network with attention is used as a decoder. In addition, Mask R-CNN with OpenCV is used for object detection and color analysis. The integration of the image caption and color recognition is then performed to provide better descriptive details of images. Moreover, the generated textual sentence is converted into speech. The validation results illustrate that the proposed method can provide more accurate description of images.

Original languageEnglish
Article number209
JournalApplied Sciences (Switzerland)
Volume12
Issue number1
DOIs
StatePublished - 01 01 2022

Bibliographical note

Publisher Copyright:
© 2021 by the authors. Licensee MDPI, Basel, Switzerland.

Keywords

  • Color recognition
  • Image caption
  • LSTM
  • Object detection

Fingerprint

Dive into the research topics of 'Enhanced image captioning with color recognition using deep learning methods'. Together they form a unique fingerprint.

Cite this