Training image classification model using python (pytorch)
Learn how to train an image classifier using python + pytorch. We will use a pretrained model (Resnet18) to classify animal images. The same process can be used to train a classifier for different contexts like medicine (radiology), autonomous vehicles, etc.
Download the Classifier file and extract it: https://drive.google.com/drive/folder...
Download VS Code: https://code.visualstudio.com/download
Download Python version 3.11.9: https://www.python.org/downloads/rele...
00:00 Introduction and mindset
04:50 Set up 1 (VScode + python and jupyter extensions)
09:00 Set up 2 (Creating a virtual environment and installing libraries)
12:50 Preparing the dataset
15:50 Inspecting the data
17:28 Defining/Loading Neural Net Architecture
18:50 Loss function + Optimizer
19:57 Training Loop
22:55 Visualizing performance
24:00 Training vs Validation vs Testing
27:50 Saving the weights of the model
28:16 Loading the weights and predicting on sample images