Edit model card

Model description

This repo contains the trained model for grapevine leaves image classification

Full credits go to Vu Minh Chien

Motivation: The main product of grapevines is grapes that are consumed fresh or processed. In addition, grapevine leaves are harvested once a year as a by-product. The species of grapevine leaves are important in terms of price and taste. In this repo, deep learning-based classification is conducted by using images of grapevine leaves

Intended uses & limitations

Images of 500 vine leaves belonging to 5 species were taken with a special self-illuminating system. Later, this number was increased to 2500 with data augmentation methods

Training and evaluation data

Training hyperparameters

The following hyperparameters were used during training:

Hyperparameters Value
name Adam
learning_rate e-3
freeze_epochs 3
unfreeze_epochs 10
training_precision float16
Downloads last month
0

Space using hugginglearners/grapevine_leaves_classification 1