Automatic-Image-Captioning
PublicA neural network architecture(CNN+LSTM) that automatically generates captions from the images. The model uses ResNet architecture to train the Encoder while DecoderRNN has to be trained with our choice of trainable parameters. I have trained the model on the Microsoft Common Objects in COntext (MS COCO) dataset and have tested the network on fictitious images!
cnndeep-learningimage-captioningimage-classificationimage-processingmachine-learningneural-networkpytorchrnn
Creat：2020-01-13T23:09:41
Update：2024-08-21T16:09:16
25
Stars
0
Stars Increase