Rosemary Disease Classifier
A fine-tuned google/vit-base-patch16-224 that classifies the condition of rosemary
(Rosmarinus officinalis) leaves. Built for the AgroSense rosemary-monitoring app.
Classes
Rosemary, Rosemary_Leaf_Spot, Rosemary_Pest_Damage, Rosemary_Powdery_Mildew
Training
- Base model: google/vit-base-patch16-224
- Epochs: 12
- Batch size: 16
- Learning rate: 2e-05
- Augmentation: random resized crop, horizontal flip, small rotation
- Splits: dataset-provided train / validation / test
Evaluation (held-out test split)
- Accuracy: 0.9014
- Macro-F1: 0.9121
Limitations
Trained on a custom rosemary image set; works best on clear, well-lit photos of
a single leaf or plant. It only knows the four classes listed above.
