Skip to content

Commit

Permalink
Update README to enhance clarity and add model evaluation results
Browse files Browse the repository at this point in the history
  • Loading branch information
Abraham KOLOBOE committed Oct 19, 2024
1 parent ea139d3 commit e8e5edc
Showing 1 changed file with 15 additions and 10 deletions.
25 changes: 15 additions & 10 deletions Readme-1.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
## **Problème de Classification** 🍎🍌🍇
Le projet vise à développer plusieurs modèles d’intelligence artificielle capables de classifier des images de fruits (100x100 pixels). L’objectif est d’entraîner des modèles pour reconnaître et classer différents types de fruits, comme des pommes, des bananes, etc.

---

Le but de ce projet est de développer et d'évaluer plusieurs modèles d'intelligence artificielle capables de classifier des images de fruits. Les images utilisées dans ce projet sont de taille 100x100 pixels. Nous cherchons à entraîner des modèles qui peuvent reconnaître et classer différents types de fruits, tels que des pommes, des bananes, etc.

Expand All @@ -20,7 +19,7 @@ Dans ce projet, nous visons à entraîner et évaluer quatre modèles différent
3. **ResNet** 🕸️
4. **VGG16** 🏛️

---


# **Flux de Travail** 🔄

Expand All @@ -46,7 +45,7 @@ Dans ce projet, nous visons à entraîner et évaluer quatre modèles différent
7. **Gestion des Artéfacts** 💾
- Sauvegarde des modèles, graphiques et journaux.

---


# **Récupérer les Données et Installer les Dépendances**

Expand All @@ -56,7 +55,6 @@ Dans ce projet, nous visons à entraîner et évaluer quatre modèles différent
2. **Installer les dépendances** :
- Installation des bibliothèques nécessaires à l’entraînement.

---

# **Paramètres Généraux du Training** 📚⚙️

Expand All @@ -65,7 +63,7 @@ Dans ce projet, nous visons à entraîner et évaluer quatre modèles différent
- **epochs** : 10
- **patience** : 2

---


# **Fonctions Utiles** 🛠️

Expand All @@ -82,20 +80,20 @@ Dans ce projet, nous visons à entraîner et évaluer quatre modèles différent
4. **Gestion des Artéfacts** 💾
- **`zip_directory()`** : Compresse un répertoire.

---


# **Chargement des Données et Visualisation des Fruits** 🍎🍌🍇

Visualisez une grille d'images issues du dataset pour vérifier leur intégrité :
![Image](image/sample-base.png)

---


# **Data Augmentation et Visualisation des Fruits Augmentés** 📈
L'augmentation inclut la rotation, le flip et l’ajustement de contraste :
![Image](image/sample-augmented.png)

---


# **Architecture des Modèles** 🏛️

Expand Down Expand Up @@ -124,7 +122,6 @@ L'augmentation inclut la rotation, le flip et l’ajustement de contraste :
</table>
</div>

---

# **Train, Plot, and Evaluate the Models** 🚀📊

Expand Down Expand Up @@ -158,7 +155,15 @@ L'augmentation inclut la rotation, le flip et l’ajustement de contraste :
| Temps d'inférence moyen ⏱️ | 0.2 |


---
| Modèle | Accuracy 🎯 | AUC 📈 | Précision 🧮 | Rappel 🔍 | F1 🏆 | Temps d'inférence moyen ⏱️ | Score final 🏅 |
|--------------|-------------|--------|--------------|-----------|-------|-----------------------------|----------------|
| CNN | 0.985317 | 0.999915| 0.988543 | 0.985317 | 0.984813| 1.000000 | 0.989986 |
| ResNet | 1.000000 | 1.000000| 1.000000 | 1.000000 | 1.000000| 0.185564 | 0.837113 |
| VGG16 | 0.974489 | 0.999853| 0.977447 | 0.974489 | 0.973609| 0.065214 | 0.795378 |
| EfficientNet | 0.954392 | 0.999682| 0.965315 | 0.954392 | 0.954058| 0.044032 | 0.777908 |




# **Conclusion** 🎉

Expand Down

0 comments on commit e8e5edc

Please sign in to comment.