Skip to content

Damrukesh/Image_captioning_IEEE

Repository files navigation

Image_captioning_IEEE

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

WhatsApp Image 2025-08-23 at 11 54 41_a4082c51

About

two implementations ( tensorflow and pytorch )

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages