Four-in-one deep network: image search, image captioning, similar words and similar images using a s...
The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:
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 'Flicker8kDataset' folder in the same directory as your notebook. You'll also need to download captions from here. Extract captions in the folder 'captiondatasets'.
Image caption example:
Similar words example
Similar images example
Image search example
Made by @paraschopra