beans-vit
google/vit-base-patch16-224 fine-tuned on AI-Lab-Makerere/beans
(bean-leaf disease classification) for Lab 1 of Przetwarzanie Danych w Chmurze.
Classes: angular_leaf_spot, bean_rust, healthy
Results (test split)
- Accuracy: 0.9688
- Macro F1: 0.9691
| class | precision | recall | f1-score | support |
|---|---|---|---|---|
| angular_leaf_spot | 0.9762 | 0.9535 | 0.9647 | 43 |
| bean_rust | 0.9333 | 0.9767 | 0.9545 | 43 |
| healthy | 1 | 0.9762 | 0.988 | 42 |

Training
Epochs: 3, learning rate: 5e-05, batch size: 16, device: cpu. Experiments tracked with MLflow.
Usage
from transformers import pipeline
clf = pipeline("image-classification", model="lustachowicz/beans-vit")
clf("leaf.jpg", top_k=3)