anshdavid/pytorch-image-caption

Image caption using VGG16 + LSTM

Python

Pytorch Image Captioning

Download the Flickr8k dataset
Extract to folder dataset

trainer.py for training the network.

tester.py for testing the network.

src/architecture.py for creating encoder, decoder and hooking them together.

Reference