Team Ai
Modelpublic

omaralshareef/beans-image-classifier

sourceHugging Faceapache-2.0updated 1mo agoView on Hugging Face
0likes29downloads
Model Card

Bean Leaf Disease Classifier

Fine-tuned MobileNetV3-Small on AI-Lab-Makerere/beans.

Given a photo of a bean leaf, the model predicts one of:

  • —angular_leaf_spot
  • —bean_rust
  • —healthy

Metrics

SplitAccuracyLoss
Validation (best checkpoint)95.49%0.235
Test89.84%0.303

Training ran for 5 epochs on Apple Silicon (MPS). The best checkpoint was restored before export.

Usage

python
from transformers import pipeline

clf = pipeline("image-classification", model="omaralshareef/beans-image-classifier")
print(clf("leaf.jpg"))

Training hyperparameters

  • —learning rate: 5e-5
  • —train batch size: 16
  • —eval batch size: 16
  • —epochs: 5
  • —optimizer: AdamW
  • —seed: 42

Training results

Training LossEpochStepValidation LossAccuracy
0.58741.0650.41240.8346
0.40422.01300.23520.9549
0.28053.01950.16790.9549
0.23674.02600.13700.9549
0.19335.03250.13940.9398