Awesome Open Source
Awesome Open Source



Pytorch Tutorial for Fine Tuning/Transfer Learning a Resnet for Image Classification

If you want to do image classification by fine tuning a pretrained mdoel, this is a tutorial will help you out. It shows how to perform fine tuning or transfer learning in PyTorch with your own data. It is based on a bunch of of official pytorch tutorials/examples. I felt that it was not exactly super trivial to perform in PyTorch, and so I thought I'd release my code as a tutorial which I wrote originally for my research.

Highly encourage you to run this on a new data set (read to know which format to store your data in), but for a sample dataset to start with, you can download a simple 2 class dataset from here -

All Torch and PyTorch specific details have been explained in detail in the file

Hope this tutorial helps you out! :)

Credits - This tutorial is built on top of mainly on 2 Pytorch tutorials - and

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
python (55,598
deep-learning (4,087
tutorial (988
image-classification (188
pytorch-tutorials (24