Upload README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,21 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Pneumonia Detection Model
|
2 |
+
|
3 |
+
## Model Description
|
4 |
+
This deep learning model is trained to detect signs of pneumonia from chest X-ray images. It utilizes the power of MobileNetV3 as a backbone for feature extraction, coupled with additional convolutional layers for improved accuracy.
|
5 |
+
|
6 |
+
## Purpose
|
7 |
+
The model is designed to assist healthcare professionals in screening patients for pneumonia, potentially speeding up diagnosis and treatment planning.
|
8 |
+
|
9 |
+
## Training Data Description
|
10 |
+
The model was trained on a comprehensive dataset of chest X-ray images sourced from [Kaggle's Pneumonia Dataset](https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia). The dataset comprises frontal-view X-ray images of pediatric patients, categorized into pneumonia and normal health status.
|
11 |
+
|
12 |
+
## Model Architecture
|
13 |
+
The architecture leverages a pre-trained MobileNetV3 model with the top layer removed. Custom convolutional layers are added to capture the nuances of X-ray images specific to pneumonia.
|
14 |
+
|
15 |
+
## Training Procedures
|
16 |
+
The model was trained using the Adam optimizer with a learning rate of 1e-4. We employed early stopping mechanisms to prevent overfitting, and the model's performance was validated against a separate validation set comprising 20% of the training data.
|
17 |
+
|
18 |
+
## Performance/Benchmarks
|
19 |
+
The model achieved an accuracy of 93.22% on the training set and 91.48% on the validation set. Further details on the model's performance, including precision, recall, and F1 score, are provided within the model's training logs.
|
20 |
+
|
21 |
+
|