ziaddBou commited on
Commit
d32d7e4
1 Parent(s): 42aa068

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -3
README.md CHANGED
@@ -1,3 +1,21 @@
1
- ---
2
- license: mit
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
+