Jupyter notebook corresponding to tutorial: One neural network network, many uses: image captioning, image search, similar images and similar words using one model
Requires following packages:
- PyTorch
- Numpy
- Matplotlib
To follow along, you'll need to download the Flickr8K dataset. Fill this form and you'll receive a download link on your email. (There's also a direct link to download the 1GB Flickr8K dataset, though not sure how long it'll stay like that). Extract the zip file in the 'Flicker8k_Dataset' folder in the same directory as your notebook. You'll also need to download captions from here. Extract captions in the folder 'caption_datasets'.
Image caption example:
Similar words example
Similar images example
Image search example
Made by @paraschopra
MIT License.