Repository: paraschopra/one-network-many-uses
Branch: master
Commit: ba4aee2a7797
Files: 4
Total size: 167.1 MB

Directory structure:
gitextract_eqq7ydyo/

├── README.md
├── four-in-one-network.ipynb
└── models/
    ├── epochs_40_loss_2_841_v6.pth
    └── inception_epochs_40.pth

================================================
FILE CONTENTS
================================================

================================================
FILE: README.md
================================================
# One neural network, many uses

Jupyter notebook corresponding to tutorial: [One neural network network, many uses: image captioning, image search, similar images and similar words using one model ](https://towardsdatascience.com/one-neural-network-many-uses-build-image-search-image-captioning-similar-words-and-similar-1e22080ce73d)

Requires following packages:
- PyTorch
- Numpy
- Matplotlib

## Dataset NOT included (please download separately)

To follow along, you'll need to download the Flickr8K dataset. Fill [this form](https://forms.illinois.edu/sec/1713398) and you'll receive a download link on your email. (There's also a [direct link to download the 1GB Flickr8K dataset](https://github.com/jbrownlee/Datasets/releases/download/Flickr8k/Flickr8k_Dataset.zip), 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](http://cs.stanford.edu/people/karpathy/deepimagesent/caption_datasets.zip). Extract captions in the folder 'caption_datasets'.

## Example performance

Image caption example:

![art 1](https://cdn-images-1.medium.com/max/1600/1*nzq72X-Umrl2DGAD5Y8GMA.png)

Similar words example

![art 2](https://cdn-images-1.medium.com/max/1600/1*tWrGWKXwWMbuocw2nXBysA.png)

Similar images example

![asrt 3](https://cdn-images-1.medium.com/max/1600/1*NZSJiMUMQi9u07oA6vI9cA.png)

Image search example

![search example](https://cdn-images-1.medium.com/max/1600/1*Sp0QuTcOoYo06MSjYsnslg.png)

![search example](https://cdn-images-1.medium.com/max/1600/1*iM8uDn3FKoj08scfL8HMTQ.png)

Made by [@paraschopra](https://twitter.com/paraschopra)

MIT License.


================================================
FILE: four-in-one-network.ipynb
================================================
[File too large to display: 42.9 MB]

================================================
FILE: models/epochs_40_loss_2_841_v6.pth
================================================
[File too large to display: 25.8 MB]

================================================
FILE: models/inception_epochs_40.pth
================================================
[File too large to display: 98.4 MB]