Skip to main content
Learn AI Create the Future — with Pranay
  • Learn
  • Paths
  • Projects
  • Practice
  • Glossary
  • Errors
  • LeetCode
  • Main site

Main site →
Select language

Translation is unavailable right now. Your browser’s own “Translate page” usually works.

Machine translation by Google

Topics

  • 🧱 Foundations 2 sections
  • 🤖 Machine Learning 11 sections
  • 📊 Statistics and Experiments 4 sections
  • 🧭 Doing the Work 9 sections
  • 🧠 Deep Learning 11 sections
  • 🧰 Libraries and Frameworks 4 sections
  • 💬 Language and NLP 15 sections
  • 👁️ Computer Vision 13 sections
    • Computer Vision 8 lessons
    • CNN Backbones and Pretraining 13 lessons
    • Training Vision Models 12 lessons
      • Overview
      • Fine-tuning a vision model
      • Freezing and unfreezing layers
      • Replacing the classifier head
      • Linear probing vs fine-tuning
      • Where ImageNet pretraining stops helping
      • Few-shot image classification
      • Domain adaptation for vision
      • Geometric augmentations
      • Mixup and CutMix
      • RandAugment and AutoAugment
      • Building an Albumentations pipeline
      • Test-time augmentation
    • Object Detection in Depth 13 lessons
    • Segmentation in Depth 11 lessons
    • Faces, People and Pose 12 lessons
    • Video Understanding and Tracking 12 lessons
    • OCR and Document Vision 11 lessons
    • 3D Vision and Depth 11 lessons
    • Image Generation and Restoration 11 lessons
    • Vision Datasets and Annotation 11 lessons
    • Evaluating Vision Models 9 lessons
    • Shipping Vision Models 9 lessons
  • 🎙️ Speech, Forecasting and Recommenders 3 sections
  • ✨ Generative AI and LLMs 4 sections
  • 🔬 Inside a Transformer 11 sections
  • ⚙️ Production and MLOps 16 sections
  • 🌍 AI in the Real World 12 sections
  • ⚖️ Safety, Ethics and Law 2 sections

See all topics

  1. Home
  2. Learn
  3. Training Vision Models

👁️ Computer Vision · Section 058

🎓 Training Vision Models

How to take a model somebody else trained and make it work on your few thousand photos.

Every lesson in this section is written by Pranay Mahendrakar.

12 of 12 lessons published · Three reading levels on every lesson

Start with “Fine-tuning a vision model”

Lessons in order

Work top to bottom. Each lesson assumes the one above it.

  1. 01 Fine-tuning a vision model
  2. 02 Freezing and unfreezing layers
  3. 03 Replacing the classifier head
  4. 04 Linear probing vs fine-tuning
  5. 05 Where ImageNet pretraining stops helping
  6. 06 Few-shot image classification
  7. 07 Domain adaptation for vision
  8. 08 Geometric augmentations
  9. 09 Mixup and CutMix
  10. 10 RandAugment and AutoAugment
  11. 11 Building an Albumentations pipeline
  12. 12 Test-time augmentation
Previous CNN Backbones and Pretraining Next Object Detection in Depth

Learn AI

Learn Artificial Intelligence the easy way.

Learn. Build. Ask. Share.

Free forever. No sign-up, no ads, no tracking.

Learn

  • All topics
  • Learning paths
  • Projects
  • Practice playground
  • AI glossary
  • Error database
  • LeetCode solutions

Start here

  • Python for AI
  • Mathematics for AI
  • Machine Learning
  • Classic Algorithms in Depth
  • Linear Models and Regularisation
  • Ensembles and Gradient Boosting

This site

  • Search
  • New lessons feed
  • Sitemap
  • About Pranay Mahendrakar
  • pranaymahendrakar.com

© 2026 Pranay Mahendrakar. Written for people who are starting from zero.

10 new lessons every day. If something here is wrong or confusing, that is worth fixing — say so.