Oxford 17 Flowers CNN

Small custom CNN trained on the Oxford 17 Flowers dataset. Classifies RGB images into 17 flower categories.

Quick start

pip install torch torchvision huggingface_hub pillow

Download the script and run it on any image:

huggingface-cli download darjja/oxford-17-flowers-cnn predict.py --local-dir .
python predict.py your_flower.jpg
python predict.py your_flower.jpg --top 5

The script downloads weights, config, and model code from this repo automatically.

Classes

Bluebell, Buttercup, Coltsfoot, Cowslip, Crocus, Daffodil, Daisy, Dandelion, Fritillary, Iris, Lilyvalley, Pansy, Snowdrop, Sunflower, Tigerlily, Tulip, Windflower