/ Images

A Neural Network to generate captions for an image

A Neural Network to generate captions for an image

Image Caption Generator

A Neural Network to generate captions for an image.

Image-Caption-Generator

Requirements

Recommended System Requirements to train model.

  • A good CPU and a GPU
  • Atleast 8gb of RAM

Required Libraries for python

  • Keras
  • Pillow
  • nltk
  • Matplotlib

Flickr8k Dataset: Dataset Request Form

Important: After downloading the dataset, put the reqired files in train_val_data folder

Procedure to Train Model

  1. Clone the repository to preserve directory structure
  2. Put the required files in train_val_data Folder
  3. Run train_val.py

Procedure to Test on new images

  1. Clone the repository to preserve directory structure
  2. Put the test image in test_data folder
  3. Run test.py

GitHub