Downloads · 30 days
22
38% of all-time downloads
vit-beans-demo
Den953/vit-beans-demo
vit-beans-demo is a image classification model from Den953. 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.
Vision Transformer (google/vit-base-patch16-224-in21k) fine-tune sur le dataset beans pour detecter les maladies foliaires du haricot.
All-time downloads
58
Public
Parameters
85.8M
774 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx430 MB · 56%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Sep 4, 2026
- Updated
- Sep 4, 2026
- SHA
- d7856e43
Base models
Datasets
Model card
From the Hugging Face model README
vit-beans-demo
Vision Transformer (google/vit-base-patch16-224-in21k) fine-tune sur le dataset
beans pour detecter les
maladies foliaires du haricot.
Realise dans le cadre du cours MLOps M2 (lab Hugging Face).
Classes
| id | label | description |
|---|---|---|
| 0 | angular_leaf_spot | tache angulaire |
| 1 | bean_rust | rouille du haricot |
| 2 | healthy | feuille saine |
Resultats
| Split | Accuracy | Loss |
|---|---|---|
| test (128 images) | 0.9609 | 0.1792 |
Utilisation
from transformers import pipeline
clf = pipeline("image-classification", model="Den953/vit-beans-demo")
clf("feuille.jpg")
Une version ONNX est disponible dans onnx/model.onnx pour l'inference hors PyTorch.
Entrainement
| Hyperparametre | Valeur |
|---|---|
| base | google/vit-base-patch16-224-in21k |
| train / test | 1034 / 128 images |
| epochs | 2 |
| learning rate | 5e-5 |
| batch size | 16 |
| optimiseur | AdamW |
| augmentation | RandomResizedCrop(224) (train), Resize + CenterCrop (eval) |
| materiel | CPU |
Entrainement realise en CPU (AMD Ryzen, 12 threads), pas sur GPU : le nombre d'epoques a ete reduit par rapport a l'enonce du lab pour tenir dans un temps raisonnable. Ce choix est indique dans le tableau d'hyperparametres ci-dessus.
Limites
Dataset de ~1 000 images d'une seule culture, prises dans des conditions de terrain specifiques (Ouganda). Le modele n'a pas ete evalue sur d'autres especes, d'autres maladies, ni sur des photos de qualite ou d'eclairage tres differents. A ne pas utiliser comme outil de diagnostic agricole reel sans validation supplementaire.
Demo
Space de demonstration : https://huggingface.co/spaces/Den953/bean-leaf-classifier
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Sep 4, 2026
- Updated
- Sep 4, 2026