Skip to content

Latest commit

 

History

History
30 lines (16 loc) · 726 Bytes

README.md

File metadata and controls

30 lines (16 loc) · 726 Bytes
Raw

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