Image Classifier from Scratch
Train a convolutional neural network to classify images into categories — start with CIFAR-10 (10 classes) or a Kaggle dataset you find interesting. The goal is to understand the full pipeline: load data, define a model, train, evaluate, and interpret results.
Skills practised
What it shows employers
You understand CNNs, can implement a training loop, and know how to evaluate a classifier beyond raw accuracy (confusion matrix, per-class precision/recall). More employers care about this than fine-tuning a pre-trained model.
Dataset
CIFAR-10 via torchvision.datasets, or any Kaggle image classification dataset
Take it further
Add a simple web interface using Gradio or Streamlit so anyone can upload an image and get a prediction.