Image Caption Generator

A Neural Network to generate captions for an image.

Examples

Credits : Towardsdatascience

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

References

• Show and Tell: A Neural Image Caption Generator - Oriol Vinyals, Alexander Toshev, Samy Bengio, Dumitru Erhan