Learn practical skills, build real-world projects, and advance your career
Updated a year ago
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
project_name='10-famous-personality-classification'import torch
import torchvision
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
matplotlib.rcParams['figure.facecolor'] = '#ffffff'
%matplotlib inlineFetching Kaggle Datasets into Google Colab
Follow the below step to download kaggle dataset. This applies to any of Kaggle dataset.