Image Classification

A classifier is a function from pixels to a probability distribution over classes. Everything else is plumbing.

Build 1.3 hr Python

Prerequisites: Phase 2 Lesson 09 (Model Evaluation), Phase 3 Lesson 10 (Mini Framework), Phase 4 Lesson 03 (CNNs)

#Learning Objectives

  • Build an end-to-end image classification pipeline on CIFAR-10: dataset, augmentation, model, training loop, evaluation
  • Explain the role of each component (dataloader, loss, optimizer, scheduler, augmentation) and predict how breaking any one of them manifests in the loss curve
  • Implement mixup, cutout, and label smoothing from scratch and justify when each is worth adding
  • Read a confusion matrix and a per-class precision/recall table to diagnose dataset and model failures beyond aggregate accuracy
Full Access lesson

Keep going. The rest of this lesson is one step away.

Unlock all 503 lessons, video, code, quizzes and every future update. Pay in your own currency with Paystack.

The text and code of this lesson are based on AI Engineering from Scratch by Rohit Ghumare, MIT licensed.