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

-

Downloads are not tracked for this model. How to track
Inference API
Drag image file here or click to browse from your device
This model can be loaded on Inference API (serverless).

Space using hugginglearners/grapevine_leaves_classification 1