Fiche Modèle pour le Transformateur de Détection de Maladies en Agriculture Intelligente

Ce modèle est un Vision Transformer (ViT) conçu pour identifier les maladies des plantes dans les cultures dans le cadre d'un système d'agriculture intelligente. Il a été entraîné sur un ensemble de données diversifié d'images de plantes, incluant différentes catégories de maladies affectant les cultures telles que le maïs, la pomme de terre, le riz et le blé. Le modèle vise à fournir aux agriculteurs et agronomes une détection de maladies en temps réel pour une meilleure gestion des cultures.

Détails du Modèle

Description du Modèle

Ce modèle Vision Transformer a été affiné pour classifier diverses maladies de plantes couramment trouvées dans les environnements agricoles. Le modèle peut classifier les maladies dans les cultures telles que le maïs, la pomme de terre, le riz et le blé, identifiant des maladies comme la rouille, le mildiou, les taches foliaires, et autres. L'objectif est de permettre l'agriculture de précision en aidant les agriculteurs à détecter les maladies précocement et prendre les mesures appropriées.

  • Développé par : Cabrel KEPSEU
  • Type de modèle : Vision Transformer (ViT)
  • Langues (NLP) : N/A (Modèle de Vision par Ordinateur)
  • Licence : Apache 2.0
  • Affiné à partir du modèle : (WinKawaks/vit-tiny-patch16-224)[https://huggingface.co/WinKawaks/vit-tiny-patch16-224]
  • Entrée : Images de cultures (format RGB)
  • Sortie : Étiquettes de classification de maladies (catégories saines ou malades)

Maladies détectées par le modèle

Culture Maladies Identifiées
Maïs Rouille Commune
Maïs Tache Grise des Feuilles
Maïs Sain
Maïs Brûlure des Feuilles
- Invalide
Pomme de terre Mildiou Précoce
Pomme de terre Saine
Pomme de terre Mildiou Tardif
Riz Tache Brune
Riz Sain
Riz Pyriculariose
Blé Rouille Brune
Blé Sain
Blé Rouille Jaune

Utilisations

Utilisation Directe

Ce modèle peut être utilisé directement pour classifier des images de cultures afin de détecter les maladies des plantes. Il est particulièrement utile pour l'agriculture de précision, permettant aux utilisateurs de surveiller la santé des cultures et prendre des interventions précoces basées sur la maladie détectée.

Utilisation en Aval

Ce modèle peut être affiné sur d'autres ensembles de données agricoles pour des cultures ou régions spécifiques afin d'améliorer ses performances ou être intégré dans des systèmes d'agriculture de précision plus larges qui incluent d'autres fonctionnalités comme les prédictions météorologiques et le contrôle d'irrigation.

Peut être quantifié ou déployé en pleine précision sur des appareils de périphérie grâce à sa petite taille de paramètres sans compromettre la précision et l'exactitude.

Utilisation Hors Portée

Ce modèle n'est pas conçu pour des tâches de classification d'images non agricoles ou pour des environnements avec des données insuffisantes ou très bruitées. L'utilisation abusive inclut l'utilisation du modèle dans des zones avec des conditions agricoles très différentes de celles sur lesquelles il a été entraîné.

Biais, Risques et Limitations

  • Le modèle peut présenter un biais envers les cultures et maladies présentes dans l'ensemble de données d'entraînement, conduisant à des performances moindres sur les maladies ou variétés de cultures non représentées.
  • Les faux négatifs (échec à détecter une maladie) peuvent résulter en des dommages de culture non traités, tandis que les faux positifs pourraient conduire à des interventions inutiles.

Recommandations

Les utilisateurs devraient évaluer le modèle sur leurs cultures spécifiques et conditions agricoles. Des mises à jour régulières et un réentraînement avec des données locales sont recommandés pour des performances optimales.

Comment Commencer avec le Modèle

from PIL import Image, UnidentifiedImageError
from transformers import ViTFeatureExtractor, ViTForImageClassification
feature_extractor = ViTFeatureExtractor.from_pretrained('cabrel09/crop_leaf_diseases_detector')
model = ViTForImageClassification.from_pretrained(
    'cabrel09/crop_leaf_diseases_detector',
    ignore_mismatched_sizes=True
)
image = Image.open('<chemin_image>')
inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs)
logits = outputs.logits
predicted_class_idx = logits.argmax(-1).item()
print("Classe prédite:", model.config.id2label[predicted_class_idx])

Détails d'Entraînement

Données d'Entraînement

Le modèle a été entraîné sur un ensemble de données contenant des images de diverses cultures avec des maladies étiquetées, incluant les catégories suivantes :

  • Maïs : Rouille Commune, Tache Grise des Feuilles, Brûlure des Feuilles, Sain
  • Pomme de terre : Mildiou Précoce, Mildiou Tardif, Saine
  • Riz : Tache Brune, Hispa, Pyriculariose, Sain
  • Blé : Rouille Brune, Rouille Jaune, Sain

L'ensemble de données inclut également des images capturées sous diverses conditions d'éclairage, à partir d'environnements contrôlés et non contrôlés et d'angles différents, pour simuler des scénarios agricoles du monde réel. Nous avons utilisé des ensembles de données publiquement disponibles, ainsi que nos propres données privées.

Procédure d'Entraînement

Le modèle a été affiné en utilisant une architecture vision transformer pré-entraînée sur l'ensemble de données ImageNet. L'ensemble de données a été prétraité en redimensionnant les images et normalisant les valeurs de pixels.

Hyperparamètres d'Entraînement

  • Taille de lot : 32
  • Taux d'apprentissage : 2e-5
  • Époques : 4
  • Optimiseur : AdamW
  • Précision : fp16

Évaluation

Matrice de confusion

Données de Test, Facteurs et Métriques

Le modèle a été évalué en utilisant un ensemble de validation composé de 20% de l'ensemble de données original, avec les métriques suivantes :

  • Précision : 98%
  • Précision (Precision) : 97%
  • Rappel : 97%
  • Score F1 : 96%

Impact Environnemental

Les émissions de carbone pendant l'entraînement du modèle peuvent être estimées en utilisant le calculateur d'impact de l'apprentissage automatique.

  • Type de matériel : NVIDIA L40S
  • Heures utilisées : 1 heure
  • Fournisseur cloud : Lightning AI

Spécifications Techniques

Architecture et Objectif du Modèle

Le modèle utilise une architecture Vision Transformer pour apprendre les représentations d'images et les classifier en catégories de maladies. Son mécanisme d'auto-attention lui permet de capturer l'information contextuelle globale dans les images, le rendant adapté à la détection de maladies agricoles.

Infrastructure de Calcul

Logiciel

  • Python 3.9
  • PyTorch 2.4.1+cu121
  • pytorch_lightning
  • Bibliothèque Transformers par Hugging Face

Citation

Si vous utilisez ce modèle dans vos recherches ou applications, veuillez le citer comme suit :

BibTeX :

@misc{cabrel2025cropdiseases,
  title={Détecteur de maladies des feuilles de culture},
  author={Cabrel Kepseu},
  year={2025},
  publisher={Hugging Face},
}

APA :

Cabrel, K. (2025). Détecteur de maladies des feuilles de culture. Hugging Face.

Contact de la Fiche Modèle

Pour plus d'informations, contactez : [email protected]

Downloads last month
10
Safetensors
Model size
5.53M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 1 Ask for provider support

Model tree for cabrel09/crop_leaf_disease_detector

Quantized
(4)
this model

Space using cabrel09/crop_leaf_disease_detector 1