Skip to content

Latest commit

 

History

History
15 lines (12 loc) · 692 Bytes

Readme.md

File metadata and controls

15 lines (12 loc) · 692 Bytes

Neural Caption Generator with Attention

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