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

Split Accuracy Loss
Validation (best checkpoint) 95.49% 0.235
Test 89.84% 0.303

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

Usage

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 Loss Epoch Step Validation Loss Accuracy
0.5874 1.0 65 0.4124 0.8346
0.4042 2.0 130 0.2352 0.9549
0.2805 3.0 195 0.1679 0.9549
0.2367 4.0 260 0.1370 0.9549
0.1933 5.0 325 0.1394 0.9398
Downloads last month
-
Safetensors
Model size
1.53M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for omaralshareef/beans-image-classifier

Finetuned
(11)
this model

Dataset used to train omaralshareef/beans-image-classifier

Evaluation results