Skip to content

susantabiswas/image-summarizer

Repository files navigation

HitCount

Image Summarization

Image summary generation or caption generation for images using CNN-LSTM network. The model used MSCOCO dataset for training the model. The entire implementation is done in Keras.

Model Performance on Test Images:

The model was trained for 22 epochs. The results are quite convincing.

Model Architecture

References:

MSCOCO dataset was downloaded from here.
Sparse entropy loss function bug fix from here

About

Caption generation or image summarization for images using LSTM network. Implementation done in keras.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published