/flower-classification-exercise

A simple image classification task, using the 17 Category Flower Dataset.

Primary LanguageJupyter NotebookMIT LicenseMIT

flower-classification-exercise

A simple image classification task, using the 17 Category Flower Dataset.

Dataset

https://www.robots.ox.ac.uk/~vgg/data/flowers/17/index.html
Created by Maria-Elena Nilsback and Andrew Zisserman.

Download the 17flowers.tgz archive and extract the images into ./data/jpg in the directory where the notebook is located.

Dependencies

  • Python 3.8
  • PyTorch
  • Torchvision
  • Cuda 11

Full Anaconda environment specification available in the environment.yaml file.