-
Notifications
You must be signed in to change notification settings - Fork 0
devakar771/SOS_Final
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
Implemeted the Encoder-Decoder mechanism of Image Captioning using Attention Mechnism. The model was trained on the Flickr 8k dataset. # The encoder model uses InceptionV3 as base model to extract features from the images. # The attention part links up the extracted features with the captions of the image. # Decoder is implemented using GRU. # Model trained weights can be found here. https://drive.google.com/file/d/11mk5t6pCyaSjAExNr7j9W9_83hLUfK6Y/view?usp=sharing The model obtained an average BLEU score of 0.57 on the validation data of 7000 captions.
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published