vinjk / image_captioning Goto Github PK
View Code? Open in Web Editor NEWDesign and train a deep neural network to automatically generate caption for an input image. PyTorch library was used to build the network and it was trained using MS COCO dataset.