Abouba1810 commited on
Commit
d34a0ef
·
verified ·
1 Parent(s): 8d1a996

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -0
README.md CHANGED
@@ -1,3 +1,42 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ tags:
4
+ - agriculture
5
+ - plant-disease-detection
6
+ - computer-vision
7
+ - efficientnet
8
+ - mali
9
  ---
10
+
11
+ # AgriMali-EfficientNetV2
12
+
13
+ AgriMali is an AI-powered agricultural assistant designed to help farmers in Mali 🇲🇱 by providing instant diagnosis for plant diseases.
14
+
15
+ ## Overview
16
+ This model is based on the **EfficientNetV2-S** architecture. It has been fine-tuned to classify plant diseases with high accuracy, ensuring robustness and efficiency for mobile-based field applications.
17
+
18
+ ## Performance Metrics
19
+ The model has been validated on a high-quality dataset, achieving the following results:
20
+
21
+ | Metric | Value |
22
+ | :--- | :--- |
23
+ | **Validation Accuracy** | **98.48%** |
24
+ | **Macro F1-Score** | 97.99% |
25
+ | **Macro AUC** | 99.98% |
26
+
27
+ ## How to use
28
+ You can easily load this model in your Python application using the `huggingface_hub` library:
29
+
30
+ ```python
31
+ from huggingface_hub import hf_hub_download
32
+ import torch
33
+
34
+ # Download the model weights
35
+ model_path = hf_hub_download(repo_id="Abouba1810/AgriMali-EfficientNetV2", filename="agrimali_best.pth")
36
+
37
+ # Initialize the model architecture (AgriMaliNet)
38
+ model = AgriMaliNet(num_classes=38)
39
+
40
+ # Load the weights
41
+ model.load_state_dict(torch.load(model_path, map_location='cpu'))
42
+ model.eval()