issues
search
jazzsaxmafia
/
show_attend_and_tell.tensorflow
BSD 2-Clause "Simplified" License
506
stars
191
forks
source link
readme
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