Learn practical skills, build real-world projects, and advance your career

Classifying 10 Famous Personality Image Dataset using Residual Network in PyTorch

A.K.A. Training an image classifier from scratch to over 90% accuracy in around 1 minute on a single GPU

In this project, we'll use the following techniques to train a state-of-the-art model in around 1 minute to achieve over 90% accuracy in classifying images from the 10 Famous personality Image Dataset,

  • Data normalization
  • Data augmentation
  • Residual connections
  • Batch normalization
  • Learning rate scheduling
  • Weight Decay
  • Gradient clipping
  • Adam optimizer
import torch
import torchvision
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
matplotlib.rcParams['figure.facecolor'] = '#ffffff'

%matplotlib inline

Fetching Kaggle Datasets into Google Colab

Follow the below step to download kaggle dataset. This applies to any of Kaggle dataset.