Skip to content

Mobilenet features and LSTMs for Image Captioning - Using TF and TF-HUB

License

Notifications You must be signed in to change notification settings

piyush2896/Image-Captioning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Mobilenet features and LSTMs for Image Captioning - Using TF and TF-HUB

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published