diff --git a/README.md b/README.md index ecec896..7ed27c7 100644 --- a/README.md +++ b/README.md @@ -26,6 +26,14 @@ > ```python > import efficientnet.tfkeras > ``` +> Or clone the project and use `Utils.FixedDropout`: +> ```python +> from Utils.FixedDropout import FixedDropout +> from keras.models import load_model +> +> # Load the model +> model = load_model('PAI_model_T.h5', custom_objects={'FixedDropout': FixedDropout}) +> ``` 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 @@ -40,14 +48,7 @@ Interface\GUI ![Example](doc/Other/CLI_V0.8.9.3.png) ### Example Image of the GUI (V0.8.9.6) โคต ![Example](doc/Other/GUI_V0.8.9.6.png) -#### __If you want to use the model in your project:__ -```python -from Utils.FixedDropout import FixedDropout -from keras.models import load_model -# Load the model -model = load_model('PAI_model_T.h5', custom_objects={'FixedDropout': FixedDropout}) -``` ## Release > ### Newest release ๐Ÿ“ƒ > #### [Go to newest release](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/releases/latest) @@ -72,7 +73,7 @@ This combined dataset provides a comprehensive set of images for training the mo ### Model list: | Model | Base Model | Params | acc | Status | |----------|-----------------|--------|--------|--------| -| V6 | efficientnet-b7 | 65.4M | 97.12% | โš ๏ธ | +| V6 | efficientnet-b7 | 65.4M | 97.12% | โœ… | | V7 light | efficientnet-b4 | 29.7M | 97.12% | โœ… | | V8 Super light | efficientnet-b0 | 4.8M | 96.47% | โœ… | diff --git a/SECURITY.md b/SECURITY.md index ae776ca..c58722d 100644 --- a/SECURITY.md +++ b/SECURITY.md @@ -3,7 +3,7 @@ ## Model list: | Model | Base Model | Params | acc | Status | |----------|-----------------|--------|--------|--------| -| V6 | efficientnet-b7 | 65.4M | 97.12% | โš ๏ธ | +| V6 | efficientnet-b7 | 65.4M | 97.12% | โœ… | | V7 light | efficientnet-b4 | 29.7M | 97.12% | โœ… | | V8 Super light | efficientnet-b0 | 4.8M | 96.47% | โœ… |