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

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

Primary LanguageJupyter NotebookMIT LicenseMIT

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