jeffrey1hu/show_attend_and_tell.tensorflow

Neural Caption Generator with Attention

• Tensorflow implementation of "Show, attend and Tell" http://arxiv.org/abs/1502.03044
• Borrowed most of the idea from the author's source code https://github.com/kelvinxu/arctic-captions

Code

• make_flickr_dataset.py: Extracts conv5_3 layer activations of VGG Network for flickr30k images, and save them in 'data/feats.npy'
• model_tensorflow.py: Main codes

Usage

• Download flickr30k Dataset.
• Extract VGG conv5_3 features using make_flickr_dataset.py
• Train: run train() in model_tensorflow.py
• Test: run test() in model_tensorflow.py