A Neural Network to generate captions for an image.
Examples
Credits : Towardsdatascience
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
- Clone the repository to preserve directory structure
- Put the required files in train_val_data Folder
- Run train_val.py
- Clone the repository to preserve directory structure
- Put the test image in test_data folder
- Run test.py
- Show and Tell: A Neural Image Caption Generator - Oriol Vinyals, Alexander Toshev, Samy Bengio, Dumitru Erhan