CoolFace
Modelpublic

omaralshareef/beans-image-classifier

sourceHugging Faceapache-2.0updated 20d agoView on Hugging Face
0likes35downloads
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