Upload any image and Neural Network will generate caption for it.
You can try it here
NodeJS is wrapped around Pytorch Neural Network.
Convolutional Neural Network is pretrained Resnet50.
CNN-RNN encoder-decoder pair was trained with COCO dataset.
Approximate process of traing is described in this paper
- Node JS, Pytorch, Numpy, PIL.
To run the app you just need to install all dependencies and hit 'node server.js'