Skip to content

Automatic image caption generator trained on flickr-8k data-set

License

Notifications You must be signed in to change notification settings

skrstv123/AUTO-IMAGE-DESCRIPTION

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Automatic image caption generator

This deep learning model is trained on the flickr_8k dataset. It uses n-gram model for text prediction and InceptionV3 for generating image embeddings. I have also used glove 6B 200d word embeddings .

performance on images from flickr-8k data-set

N|Solid

N|Solid

performance on real world images

N|Solid

N|Solid

The trained model and tokenizers can be found inside the assets folder.

License

MIT

Free Software, Hell Yeah!

About

Automatic image caption generator trained on flickr-8k data-set

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages