Davidsv commited on
Commit
0570906
·
verified ·
1 Parent(s): 0d83b76

Update model card with examples and purpose

Browse files
Files changed (1) hide show
  1. README.md +91 -17
README.md CHANGED
@@ -8,6 +8,8 @@ tags:
8
  - computer-vision
9
  - object-detection
10
  - rf-detr
 
 
11
  datasets:
12
  - plant-disease-faxnj
13
  metrics:
@@ -17,27 +19,89 @@ pipeline_tag: object-detection
17
 
18
  # CropScan - Plant Disease Detection Model
19
 
20
- CropScan is an RF-DETR model fine-tuned for plant disease detection. It identifies and localizes disease regions on plant leaves.
21
 
22
- ## Model Details
23
 
24
- - **Model Architecture**: RF-DETR (medium)
25
- - **Task**: Object Detection / Disease Localization
26
- - **Performance**: mAP@50: 0.502
27
- - **Training**: Fine-tuned on plant disease dataset
28
 
29
- ## Usage
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
 
31
  ```python
 
32
  from rfdetr import RFDETRBase
 
33
 
 
34
  model = RFDETRBase()
35
- model.load_state_dict(torch.load("checkpoint_best_total.pth"))
 
 
 
 
 
 
 
 
 
 
 
36
  ```
37
 
38
- ## Training Data
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
 
40
- This model was trained on the Plant Disease dataset from Roboflow Universe.
 
 
41
 
42
  ```bibtex
43
  @misc{plant-disease-faxnj_dataset,
@@ -53,14 +117,24 @@ This model was trained on the Plant Disease dataset from Roboflow Universe.
53
  }
54
  ```
55
 
56
- ## Intended Use
57
 
58
- - Plant disease detection in agricultural applications
59
- - Research on plant pathology
60
- - Integration with segmentation models (e.g., SAM2) for precise mask generation
 
61
 
62
  ## Limitations
63
 
64
- - Trained primarily on PlantVillage-style images
65
- - Best performance on individual leaf images with clear backgrounds
66
- - May require SAM2 integration for precise segmentation masks
 
 
 
 
 
 
 
 
 
 
8
  - computer-vision
9
  - object-detection
10
  - rf-detr
11
+ - precision-agriculture
12
+ - crop-health
13
  datasets:
14
  - plant-disease-faxnj
15
  metrics:
 
19
 
20
  # CropScan - Plant Disease Detection Model
21
 
22
+ CropScan est un modele de detection de maladies des plantes base sur RF-DETR, concu pour aider les agriculteurs a identifier rapidement les problemes de sante de leurs cultures.
23
 
24
+ ## Pourquoi CropScan ?
25
 
26
+ L'agriculture est un metier difficile. Les agriculteurs font face a de nombreux defis quotidiens : conditions meteorologiques imprevisibles, pressions economiques, et surtout, les maladies des cultures qui peuvent devastar des recoltes entieres en quelques jours.
 
 
 
27
 
28
+ **CropScan a ete cree pour :**
29
+
30
+ - **Aider les agriculteurs** a detecter les maladies tot, avant qu'elles ne se propagent
31
+ - **Reduire les pertes de recolte** grace a une intervention rapide et ciblee
32
+ - **Optimiser l'utilisation des traitements** en identifiant precisement les zones affectees
33
+ - **Democratiser l'acces** a des outils de diagnostic avances, autrefois reserves aux experts
34
+
35
+ Que vous soyez un petit exploitant ou un grand producteur, CropScan vous donne les moyens de proteger vos cultures avec la puissance de l'intelligence artificielle.
36
+
37
+ ## Exemple de Detection
38
+
39
+ | Image Originale | Resultat de Detection |
40
+ |:---------------:|:---------------------:|
41
+ | ![Original](examples/original.png) | ![Detection](examples/detection_result.png) |
42
+
43
+ L'image de gauche montre une feuille avec des symptomes de maladie. L'image de droite montre le resultat de CropScan : chaque region malade est identifiee et segmentee avec precision grace a l'integration de SAM2.
44
+
45
+ ## Details Techniques
46
+
47
+ | Specification | Valeur |
48
+ |--------------|--------|
49
+ | **Architecture** | RF-DETR (medium) |
50
+ | **Tache** | Detection d'objets / Localisation de maladies |
51
+ | **Performance** | mAP@50: 0.502 |
52
+ | **Taille du modele** | 134 MB |
53
+ | **Format** | PyTorch (.pth) |
54
+
55
+ ## Utilisation
56
+
57
+ ### Installation
58
+
59
+ ```bash
60
+ pip install rfdetr torch torchvision
61
+ ```
62
+
63
+ ### Inference
64
 
65
  ```python
66
+ import torch
67
  from rfdetr import RFDETRBase
68
+ from PIL import Image
69
 
70
+ # Charger le modele
71
  model = RFDETRBase()
72
+ checkpoint = torch.load("checkpoint_best_total.pth", map_location="cpu")
73
+ model.load_state_dict(checkpoint)
74
+ model.eval()
75
+
76
+ # Charger une image
77
+ image = Image.open("votre_image.jpg")
78
+
79
+ # Effectuer la detection
80
+ with torch.no_grad():
81
+ predictions = model(image)
82
+
83
+ # Les predictions contiennent les boites englobantes des zones malades
84
  ```
85
 
86
+ ### Integration avec SAM2 (Recommande)
87
+
88
+ Pour obtenir des masques de segmentation precis plutot que des boites englobantes, combinez CropScan avec SAM2 :
89
+
90
+ ```python
91
+ from sam2.sam2_image_predictor import SAM2ImagePredictor
92
+
93
+ # Utiliser les boites de CropScan comme prompts pour SAM2
94
+ predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-small")
95
+ predictor.set_image(image)
96
+
97
+ for box in predictions.boxes:
98
+ masks, scores, _ = predictor.predict(box=box, multimask_output=False)
99
+ # masks contient le masque de segmentation precis
100
+ ```
101
 
102
+ ## Donnees d'Entrainement
103
+
104
+ Ce modele a ete entraine sur le dataset Plant Disease de Roboflow Universe, contenant des images de feuilles avec diverses maladies.
105
 
106
  ```bibtex
107
  @misc{plant-disease-faxnj_dataset,
 
117
  }
118
  ```
119
 
120
+ ## Cas d'Usage
121
 
122
+ - **Agriculture de precision** : Surveillance automatisee des cultures par drone ou camera fixe
123
+ - **Diagnostic sur le terrain** : Application mobile pour identification rapide des maladies
124
+ - **Recherche agronomique** : Etude de la propagation des maladies vegetales
125
+ - **Formation** : Outil pedagogique pour les etudiants en agronomie
126
 
127
  ## Limitations
128
 
129
+ - Entraine principalement sur des images de type PlantVillage
130
+ - Meilleures performances sur des images individuelles de feuilles avec fond clair
131
+ - Recommande d'utiliser SAM2 pour des masques de segmentation precis
132
+ - Ne remplace pas l'expertise d'un agronome pour le diagnostic final
133
+
134
+ ## Licence
135
+
136
+ Ce modele est distribue sous licence MIT. Vous etes libre de l'utiliser, le modifier et le distribuer pour des usages commerciaux ou non-commerciaux.
137
+
138
+ ---
139
+
140
+ *Cree avec passion pour soutenir ceux qui nous nourrissent.*