Trained an EncoderCNN-DecoderRNN model to automatically generate captions for given image using COCO dataset. ResNet50 to generate feature vectors from images along with embeddings for captions to train LSTM
Basic Architecture Diagram
anirbansen2709 / imagecaptioningusingcocodataset Goto Github PK
View Code? Open in Web Editor NEWImageCaptioningUsingCOCODataset