Skip to content
Back to course

13.2 Image Classification

Image classification answers one question: what is in this picture? The model takes a whole image and outputs a single label such as 'cat', 'injera' or 'invoice', usually with a confidence score. Modern classifiers are convolutional neural networks (CNNs) that learn layers of features, from simple edges in early layers to whole objects in deeper layers.

A CNN stacks convolution and pooling layers, ending in a classifier that outputs class probabilities.

A coffee exporter could train a classifier to sort photos of beans into grades. The model never says where each defect is; it just gives the overall label, like 'Grade 1' or 'Grade 3', for the whole image of a sample.

Scenario

You must label each whole X-ray as 'normal' or 'pneumonia', without marking where the problem is. Which task fits?

Check your understanding

1/4 · 40 XP

What does an image classifier output?