Downloads · 30 days
36
100% of all-time downloads
omaralshareef/beans-image-classifier
beans-image-classifier is a image classification model from omaralshareef. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Fine-tuned MobileNetV3-Small on AI-Lab-Makerere/beans.
Downloads · 30 days
36
100% of all-time downloads
All-time downloads
36
Public
Parameters
1.5M
6.2 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors6.2 MB · 100%
From the Hugging Face model README
Fine-tuned MobileNetV3-Small on AI-Lab-Makerere/beans.
Given a photo of a bean leaf, the model predicts one of:
angular_leaf_spotbean_rusthealthy| 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.
from transformers import pipeline
clf = pipeline("image-classification", model="omaralshareef/beans-image-classifier")
print(clf("leaf.jpg"))
5e-5| 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 |