Image Captioning

This Repository Contains code for image captioning. The code uses TensorFlow Hub mobilenet module for image features. These features are given as hidden state input to the LSTM Cells. The code here is inspired by @hvass-labs TensorFlow tutorials. But is changed a little for experimentation.

Model

Image Caption using mobilenet v1

Note: This picture was made by editing @hvass-labs TensorFlow tutorials Image captioning image.

Dependencies

1. PIL
2. TensorFlow
3. TensorFlow-Hub
4. NumPy
5. Matplotlib