Skip to content

Automatically generates captions for an image using Image processing and NLP. Model was trained on Flickr30K dataset.

License

Notifications You must be signed in to change notification settings

nithintata/image-caption-generator-using-deep-learning

Repository files navigation

License: MIT Python CodeScene Code Health

Image-Caption-Generator-Using-Deep-Learning

A Deep Neural Network based generative model for captioning images using CNN and LSTM with BEAM Search..

Automatically generates captions for an image using CNN and NLP. Model was trained on Flickr30K dataset.

Kaggle kernel: https://www.kaggle.com/ntikmr/image-captioning

Steps to install

  • Clone the entire repository into a new folder.
  • Install the required dependencies using pip
  • Open gui.py and run the file

Pre-Requisites

  • Tensorflow V2.0
  • pandas
  • numpy
  • pickle
  • PIL
  • OpenCv

Examples