cabrel09 commited on
Commit
abe1540
·
verified ·
1 Parent(s): 29bb966

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -1,35 +1,35 @@
1
- *.7z filter=lfs diff=lfs merge=lfs -text
2
- *.arrow filter=lfs diff=lfs merge=lfs -text
3
- *.bin filter=lfs diff=lfs merge=lfs -text
4
- *.bz2 filter=lfs diff=lfs merge=lfs -text
5
- *.ckpt filter=lfs diff=lfs merge=lfs -text
6
- *.ftz filter=lfs diff=lfs merge=lfs -text
7
- *.gz filter=lfs diff=lfs merge=lfs -text
8
- *.h5 filter=lfs diff=lfs merge=lfs -text
9
- *.joblib filter=lfs diff=lfs merge=lfs -text
10
- *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
- *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
- *.model filter=lfs diff=lfs merge=lfs -text
13
- *.msgpack filter=lfs diff=lfs merge=lfs -text
14
- *.npy filter=lfs diff=lfs merge=lfs -text
15
- *.npz filter=lfs diff=lfs merge=lfs -text
16
- *.onnx filter=lfs diff=lfs merge=lfs -text
17
- *.ot filter=lfs diff=lfs merge=lfs -text
18
- *.parquet filter=lfs diff=lfs merge=lfs -text
19
- *.pb filter=lfs diff=lfs merge=lfs -text
20
- *.pickle filter=lfs diff=lfs merge=lfs -text
21
- *.pkl filter=lfs diff=lfs merge=lfs -text
22
- *.pt filter=lfs diff=lfs merge=lfs -text
23
- *.pth filter=lfs diff=lfs merge=lfs -text
24
- *.rar filter=lfs diff=lfs merge=lfs -text
25
- *.safetensors filter=lfs diff=lfs merge=lfs -text
26
- saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
- *.tar.* filter=lfs diff=lfs merge=lfs -text
28
- *.tar filter=lfs diff=lfs merge=lfs -text
29
- *.tflite filter=lfs diff=lfs merge=lfs -text
30
- *.tgz filter=lfs diff=lfs merge=lfs -text
31
- *.wasm filter=lfs diff=lfs merge=lfs -text
32
- *.xz filter=lfs diff=lfs merge=lfs -text
33
- *.zip filter=lfs diff=lfs merge=lfs -text
34
- *.zst filter=lfs diff=lfs merge=lfs -text
35
- *tfevents* filter=lfs diff=lfs merge=lfs -text
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,179 @@
1
  ---
2
- license: apache-2.0
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model:
3
+ - WinKawaks/vit-tiny-patch16-224
4
+ library_name: transformers
5
+ license: mit
6
+ metrics:
7
+ - accuracy
8
+ pipeline_tag: image-classification
9
+ tags:
10
+ - vision transformer
11
+ - agriculture
12
+ - plant disease detection
13
+ - smart farming
14
+ - image classification
15
  ---
16
+
17
+ # Fiche Modèle pour le Transformateur de Détection de Maladies en Agriculture Intelligente
18
+
19
+ 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.
20
+
21
+ ## Détails du Modèle
22
+
23
+ ### Description du Modèle
24
+
25
+ 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.
26
+
27
+ - **Développé par :** Cabrel KEPSEU
28
+ - **Type de modèle :** Vision Transformer (ViT)
29
+ - **Langues (NLP) :** N/A (Modèle de Vision par Ordinateur)
30
+ - **Licence :** Apache 2.0
31
+ - **Affiné à partir du modèle :** (WinKawaks/vit-tiny-patch16-224)[https://huggingface.co/WinKawaks/vit-tiny-patch16-224]
32
+ - **Entrée :** Images de cultures (format RGB)
33
+ - **Sortie :** Étiquettes de classification de maladies (catégories saines ou malades)
34
+ ## Maladies détectées par le modèle
35
+
36
+ | Culture | Maladies Identifiées |
37
+ |---------|------------------------------|
38
+ | Maïs | Rouille Commune |
39
+ | Maïs | Tache Grise des Feuilles |
40
+ | Maïs | Sain |
41
+ | Maïs | Brûlure des Feuilles |
42
+ | - | Invalide |
43
+ | Pomme de terre | Mildiou Précoce |
44
+ | Pomme de terre | Saine |
45
+ | Pomme de terre | Mildiou Tardif |
46
+ | Riz | Tache Brune |
47
+ | Riz | Sain |
48
+ | Riz | Pyriculariose |
49
+ | Blé | Rouille Brune |
50
+ | Blé | Sain |
51
+ | Blé | Rouille Jaune |
52
+
53
+
54
+
55
+ ## Utilisations
56
+
57
+ ### Utilisation Directe
58
+
59
+ 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.
60
+
61
+ ### Utilisation en Aval
62
+
63
+ 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.
64
+
65
+ 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.
66
+ ### Utilisation Hors Portée
67
+
68
+ 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é.
69
+
70
+ ## Biais, Risques et Limitations
71
+
72
+ - 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.
73
+ - 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.
74
+
75
+ ### Recommandations
76
+
77
+ 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.
78
+
79
+ ## Comment Commencer avec le Modèle
80
+
81
+ ```python
82
+ from PIL import Image, UnidentifiedImageError
83
+ from transformers import ViTFeatureExtractor, ViTForImageClassification
84
+ feature_extractor = ViTFeatureExtractor.from_pretrained('cabrel09/crop_leaf_diseases_detector')
85
+ model = ViTForImageClassification.from_pretrained(
86
+ 'cabrel09/crop_leaf_diseases_detector',
87
+ ignore_mismatched_sizes=True
88
+ )
89
+ image = Image.open('<chemin_image>')
90
+ inputs = feature_extractor(images=image, return_tensors="pt")
91
+ outputs = model(**inputs)
92
+ logits = outputs.logits
93
+ predicted_class_idx = logits.argmax(-1).item()
94
+ print("Classe prédite:", model.config.id2label[predicted_class_idx])
95
+ ```
96
+
97
+ ## Détails d'Entraînement
98
+
99
+ ### Données d'Entraînement
100
+
101
+ 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 :
102
+
103
+ - **Maïs** : Rouille Commune, Tache Grise des Feuilles, Brûlure des Feuilles, Sain
104
+ - **Pomme de terre** : Mildiou Précoce, Mildiou Tardif, Saine
105
+ - **Riz** : Tache Brune, Hispa, Pyriculariose, Sain
106
+ - **Blé** : Rouille Brune, Rouille Jaune, Sain
107
+
108
+ 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.
109
+ Nous avons utilisé des ensembles de données publiquement disponibles, ainsi que nos propres données privées.
110
+ ### Procédure d'Entraînement
111
+
112
+ 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.
113
+
114
+ #### Hyperparamètres d'Entraînement
115
+
116
+ - **Taille de lot :** 32
117
+ - **Taux d'apprentissage :** 2e-5
118
+ - **Époques :** 4
119
+ - **Optimiseur :** AdamW
120
+ - **Précision :** fp16
121
+
122
+ ### Évaluation
123
+ ![Matrice de confusion](disease_classification_metrics.png)
124
+
125
+
126
+ #### Données de Test, Facteurs et Métriques
127
+
128
+ 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 :
129
+
130
+ - **Précision :** 98%
131
+ - **Précision (Precision) :** 97%
132
+ - **Rappel :** 97%
133
+ - **Score F1 :** 96%
134
+
135
+ ## Impact Environnemental
136
+
137
+ 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](https://mlco2.github.io/impact#compute).
138
+
139
+ - **Type de matériel :** NVIDIA L40S
140
+ - **Heures utilisées :** 1 heure
141
+ - **Fournisseur cloud :** Lightning AI
142
+
143
+ ## Spécifications Techniques
144
+
145
+ ### Architecture et Objectif du Modèle
146
+
147
+ 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.
148
+
149
+ ### Infrastructure de Calcul
150
+
151
+ #### Logiciel
152
+
153
+ - Python 3.9
154
+ - PyTorch 2.4.1+cu121
155
+ - pytorch_lightning
156
+ - Bibliothèque Transformers par Hugging Face
157
+
158
+ ## Citation
159
+
160
+ Si vous utilisez ce modèle dans vos recherches ou applications, veuillez le citer comme suit :
161
+
162
+ **BibTeX :**
163
+
164
+ ```
165
+ @misc{cabrel2025cropdiseases,
166
+ title={Détecteur de maladies des feuilles de culture},
167
+ author={Cabrel Kepseu},
168
+ year={2025},
169
+ publisher={Hugging Face},
170
+ }
171
+ ```
172
+
173
+ **APA :**
174
+
175
+ Cabrel, K. (2025). Détecteur de maladies des feuilles de culture. Hugging Face.
176
+
177
+ ## Contact de la Fiche Modèle
178
+
179
+ Pour plus d'informations, contactez : klccomputing2023@gmail.com
config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "WinKawaks/vit-tiny-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 192,
11
+ "id2label": {
12
+ "0": "Corn___Common_Rust",
13
+ "1": "Corn___Gray_Leaf_Spot",
14
+ "10": "Wheat___Brown_Rust",
15
+ "11": "Wheat___Healthy",
16
+ "12": "Wheat___Yellow_Rust",
17
+ "2": "Corn___Healthy",
18
+ "3": "Invalid",
19
+ "4": "Potato___Early_Blight",
20
+ "5": "Potato___Healthy",
21
+ "6": "Potato___Late_Blight",
22
+ "7": "Rice___Brown_Spot",
23
+ "8": "Rice___Healthy",
24
+ "9": "Rice___Leaf_Blast"
25
+ },
26
+ "image_size": 224,
27
+ "initializer_range": 0.02,
28
+ "intermediate_size": 768,
29
+ "label2id": {
30
+ "Corn___Common_Rust": "0",
31
+ "Corn___Gray_Leaf_Spot": "1",
32
+ "Corn___Healthy": "2",
33
+ "Invalid": "3",
34
+ "Potato___Early_Blight": "4",
35
+ "Potato___Healthy": "5",
36
+ "Potato___Late_Blight": "6",
37
+ "Rice___Brown_Spot": "7",
38
+ "Rice___Healthy": "8",
39
+ "Rice___Leaf_Blast": "9",
40
+ "Wheat___Brown_Rust": "10",
41
+ "Wheat___Healthy": "11",
42
+ "Wheat___Yellow_Rust": "12"
43
+ },
44
+ "layer_norm_eps": 1e-12,
45
+ "model_type": "vit",
46
+ "num_attention_heads": 3,
47
+ "num_channels": 3,
48
+ "num_hidden_layers": 12,
49
+ "patch_size": 16,
50
+ "problem_type": "single_label_classification",
51
+ "qkv_bias": true,
52
+ "torch_dtype": "float32",
53
+ "transformers_version": "4.44.2"
54
+ }
disease_classification_metrics.png ADDED
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f91c9032e290d145bec27d8b82439bc9ea189f4c72923740daf4113990314b2
3
+ size 22130420
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }