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