katnoria / image-caption

Neural Image Caption Generators

Neural Image Caption Generator

TODO: Tidy up and add code

This repository contains the code to generate image captions using deep neural networks in TensorFlow 2.0

The code was used to generate the content/examples shown in my following posts:

1. Neural Image Caption - Part 1
2. Visualising Decoding Algorithms
3. Effect of Parameters on Beam Search

Papers

• Show and tell
• Show, attend and tell

Train

you can train the model using:

TODO: Add steps to install packages

Evaluate

you can evaluate the model using:

TODO: Add steps to load the checkpoint and evaluate