/PyTorch-Resnet-Example

Implementation of Aman Kharwal's ML tutorial

Primary LanguageJupyter Notebook

This repository is synced

Implementation of Aman Kharwal's tutorial found at:

https://thecleverprogrammer.com/2020/08/26/image-recognition-with-machine-learning-using-pytorch/

I decided to organize the presented knowledge together in a simple Jupyter notebook.

The missing dataset labels have been obtained from: https://gist.github.com/ageitgey/4e1342c10a71981d0b491e1b8227328b

ResNet-101 was trained using ImageNet dataset

More info at: https://www.mathworks.com/help/deeplearning/ref/resnet101.html

Dependencies:

  • pip install torch torchvision torchaudio
  • pip install Pillow

Remember to run: python -m pip install --upgrade pip

In case of outdated packages and/or outdated Jupyter:

  • pip install pip-review
  • pip-review --local --interactive
  • pip install --upgrade --user nbconvert