README.md

---
tags:
- fastai
- image-classification
---
## Model description
This repo contains the trained model for grapevine leaves image classification

Full credits go to [Vu Minh Chien](https://www.linkedin.com/in/vumichien/)

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 |