Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/jazzsaxmafia/show_attend_and_tell.tensorflow


https://github.com/jazzsaxmafia/show_attend_and_tell.tensorflow

Last synced: 10 days ago
JSON representation

Awesome Lists containing this project

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

![alt tag](https://github.com/jazzsaxmafia/show_attend_and_tell.tensorflow/blob/master/attend.jpg)