Update model card with examples and purpose
Browse files
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
|
| 21 |
|
| 22 |
-
##
|
| 23 |
|
| 24 |
-
|
| 25 |
-
- **Task**: Object Detection / Disease Localization
|
| 26 |
-
- **Performance**: mAP@50: 0.502
|
| 27 |
-
- **Training**: Fine-tuned on plant disease dataset
|
| 28 |
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
```python
|
|
|
|
| 32 |
from rfdetr import RFDETRBase
|
|
|
|
| 33 |
|
|
|
|
| 34 |
model = RFDETRBase()
|
| 35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
```
|
| 37 |
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
-
|
|
|
|
|
|
|
| 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 |
-
##
|
| 57 |
|
| 58 |
-
-
|
| 59 |
-
-
|
| 60 |
-
-
|
|
|
|
| 61 |
|
| 62 |
## Limitations
|
| 63 |
|
| 64 |
-
-
|
| 65 |
-
-
|
| 66 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
|  |  |
|
| 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.*
|