parksunwoo / show_attend_and_tell_pytorch Public

Notifications You must be signed in to change notification settings
Fork 30
Star 90

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

90 stars 30 forks Branches Tags Activity

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
README.md		README.md
data_loader.py		data_loader.py
main.py		main.py
model.py		model.py
prepro.py		prepro.py
utils.py		utils.py

Repository files navigation

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