GitHub - hextag/ML-based-Image-Caption-generator: Engineered an advanced ML model which generates descriptive captions for given input images. Used VGG16 for image segmentation, integrated this with an LSTM network to generate the captions. The model is trained on Flickr30k dataset. Achieved a strong BLEU score of 0.68, demonstrating the model’s effectiveness.

hextag / ML-based-Image-Caption-generator Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Engineered an advanced ML model which generates descriptive captions for given input images. Used VGG16 for image segmentation, integrated this with an LSTM network to generate the captions. The model is trained on Flickr30k dataset. Achieved a strong BLEU score of 0.68, demonstrating the model’s effectiveness.