garvitgit / caption-generator Goto Github PK
View Code? Open in Web Editor NEWFrom the input image it generates caption of the image. the model uses computer vision ,basically a pretrained vgg model which gives a feature vector of the image, and natural language precessing, a lstm model to generate sentences trained on sentences and the feature vector as the input.