L0ittle/show_attend_and_tell_pytorch

Pytorch implement Show, Attend and Tell: Neural Image Caption Generation with Visual Attention

Show Attend and Tell

Pytorch implement "Show, Attend and Tell: Neural Image Caption Generation with Visual Attention"

Reference

• author's code : https://github.com/kelvinxu/arctic-captions
  
• pytorch-tutorial : https://github.com/yunjey/pytorch-tutorial/tree/master/tutorials/03-advanced/image_captioning
  
• imageCaptioning : https://github.com/ruotianluo/ImageCaptioning.pytorch

Usage

• download dataset(COCO) : http://cocodataset.org/#download
  
• preprocessing : prepro.py
• train model : main.py