Rosemary Disease Classifier

--- language: - en license: mit tags: - image-classification - vit - agriculture - rosemary - plant-disease metrics: - accuracy - f1 pipeline_tag: image-classification --- # 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.

Source context: 2 downloads · 0 likes · Pipeline image-classification · Repo asafe51/rosemary-disease-classifier