Image Caption Generation

Automatic image caption generation using CNN and LSTM

Automatic image caption generation using CNN to condition a LSTM-based language model.

Approach

  • CNN encoder for image feature extraction
  • LSTM decoder for caption generation
  • Attention mechanism for improved captions

Resources

  • ๐Ÿ’ป Code: GitHub
  • ๐Ÿ“„ Report: PDF
  • ๐Ÿ“Š Slides: PDF