diff --git a/README.md b/README.md index c6782d2..15ffa7e 100644 --- a/README.md +++ b/README.md @@ -6,11 +6,11 @@ [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT) [![CodeQL](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/codeql.yml/badge.svg?branch=main)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/codeql.yml) [![Dependency Review](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/dependency-review.yml/badge.svg)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/dependency-review.yml)\ -[![Python application](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app.yml/badge.svg)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app.yml) -[![Python Test [Beta-b]](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Beta-b.yml/badge.svg)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Beta-b.yml)\ -[![Python Test [Alpha-b]](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Alpha-b.yml/badge.svg)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Alpha-b.yml) +[![Python Test [main]](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app.yml/badge.svg?branch=main)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app.yml) +[![Python Test [Beta-b]](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Beta-b.yml/badge.svg?branch=Beta-b)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Beta-b.yml)\ +[![Python Test [Alpha-b]](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Alpha-b.yml/badge.svg?branch=Alpha-b)](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/actions/workflows/python-app_Alpha-b.yml) -### This project uses a deep learning model built with the TensorFlow framework to detect pneumonia in X-ray images. The model architecture is based on the EfficientNetB7 model, which has achieved an accuracy of approximately 97.12% (97.11538%) on our test data. This high accuracy rate is one of the strengths of our AI model. +### This project uses a deep learning model built with the TensorFlow Library to detect pneumonia in X-ray images. The model architecture is based on the EfficientNetB7 model, which has achieved an accuracy of approximately 97.12% (97.11538%) on our test data. This high accuracy rate is one of the strengths of our AI model. > [!IMPORTANT] > The code that have achived the highest acc is `backup/V6/Model_T&T.ipynb`.\ > And the code with the light model is `backup/V7/Model_T&T.ipynb`. @@ -19,7 +19,7 @@ > [!TIP] > If you just want the model go to the Github Releases. -The project includes a Command Line Interface (CLI) for easy use of the model. The CLI, which is based on the [Python CLI template](https://github.com/Aydinhamedi/Python-CLI-template) from the same author, provides a user-friendly, colorful interface that allows you to interact with the model. you can fined the CLI in +The project includes a Command Line Interface (CLI) and a (GUI) Graphical User Interface for easy use of the model. The CLI, which is based on the [Python CLI template](https://github.com/Aydinhamedi/Python-CLI-template) from the same author, provides a user-friendly, colorful interface that allows you to interact with the model. you can fined the CLI in ``` Interface\CLI @@ -51,7 +51,13 @@ The model is a Convolutional Neural Network (CNN) trained on a dataset of 23681 - Covid19-Pneumonia-Normal Chest X-Ray Images from Mendeley - RSNA dataset -This combined dataset provides a comprehensive set of images for training the model. +This combined dataset provides a comprehensive set of images for training the model.\ + +### Model list: +| Model | Base Model | Params | acc | +|----------|-----------------|--------|--------| +| V6 | efficientnet-b7 | 65.4M | 97.12% | +| V7 light | efficientnet-b4 | 29.7M | 97.12% | ## Training Methods ### The AI model supports two distinct training approaches: @@ -102,9 +108,10 @@ The model provided in this project should not be used for medical diagnosis with ### Acc: ![img_](doc/V6/D1.png) ### Grad cam: -![img_](doc/V6+/D1.png) -![img_](doc/V6+/D2.png) -![img_](doc/V6+/D3.png) +| Model | Grad-cam Ex | +|----------|----------| +| V6 | ![img_](doc/V6+/D1.png)![img_](doc/V6+/D2.png)![img_](doc/V6+/D3.png)| +| V7 light | 🚧None🚧| ### Other: ![img_](doc/V6/D4.png)