Skip to content

Latest commit

 

History

History
19 lines (15 loc) · 655 Bytes

README.md

File metadata and controls

19 lines (15 loc) · 655 Bytes

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 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