Skip to content

A neural image caption generator based on the "Show and Tell" paper.

Notifications You must be signed in to change notification settings

nickvdw/image-caption-generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 

Repository files navigation

Image Caption Generator

In the notebook, we implement and test the image caption geneartor proposed in [1]. We use the MobileNetV2 network to generate neural codes for each image in the flickr8k dataset. We then create encoder and decoder models based on RNNs. This is followed by predicting and evaluating the test captions using a greedy search and BLEU scores, respectively.

[1] Oriol Vinyals, Alexander Toshev, Samy Bengio, Dumitru Erhan, "Show and Tell: A Neural Image Caption Generator", CVPR, 2015. https://arxiv.org/abs/1411.4555

About

A neural image caption generator based on the "Show and Tell" paper.

Topics

Resources

Stars

Watchers

Forks