Skip to content

Used deep learning to train a CNN + RNN/LSTM on the MS-COCO dataset to automatically generate captions.

Notifications You must be signed in to change notification settings

roshanr11/image-captioning

Repository files navigation

Image Captioning Project

In this project, I trained a Convolutional Neural Network (CNN) and a Recurrent Neural Network (RNN) with several stacked Long-Short Term Memory (LSTM) model layers on the MS-COCO dataset to effectively generate captions for any reasonable image.

About

Used deep learning to train a CNN + RNN/LSTM on the MS-COCO dataset to automatically generate captions.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published