Implemented two image captioning models trained on Flickr dataset [ 6000 training images ] Baseline model impleemented with Xception CNN and LSTM for decoder architecture Improved model by fusing CNN features with Scene graph embeddings obtrained from torchvision
-
Notifications
You must be signed in to change notification settings - Fork 0
Damrukesh/Image_captioning_IEEE
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
two implementations ( tensorflow and pytorch )
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
