Skip main navigation

New offer! Get 30% off one whole year of Unlimited learning. Subscribe for just £249.99 £174.99. New subscribers only. T&Cs apply

Find out more

Practical: Making an image classifier

A link to a Colab notebook demonstrating how to train a deep learning image classifier using images of different flower species.
A screenshot of a section of Colab notebook, showing some code, and an image of some Snowdrop flowers, correctly identified
© The University of Nottingham
To use new image data we will need to make a custom dataset.

In the linked Colab notebook we will put together everything we have learned so far to make a working image classifier for the Oxford flower dataset.

In the notebook we will cover the following:

  • downloading the dataset, and setting up datasets and dataloaders
  • importing a model network from PyTorch: ResNet
  • adapting the ResNet architecture for our image classifcation problem
  • training the network
  • testing and displaying network output.

Follow the link below and work through the Colab notebook step by step.

Image classifier

Please leave any questions or comments below.

This article is from the free online

Deep Learning for Bioscientists

Created by
FutureLearn - Learning For Life

Reach your personal and professional goals

Unlock access to hundreds of expert online courses and degrees from top universities and educators to gain accredited qualifications and professional CV-building certificates.

Join over 18 million learners to launch, switch or build upon your career, all at your own pace, across a wide range of topic areas.

Start Learning now