|
a |
|
b/README.md |
|
|
1 |
## Pneumonia Detection from Chest X-Ray Images |
|
|
2 |
|
|
|
3 |
<pre> |
|
|
4 |
Domain : Computer Vision, Machine Learning |
|
|
5 |
Sub-Domain : Deep Learning, Image Recognition |
|
|
6 |
Techniques : Deep Convolutional Neural Network, ImageNet, Inception |
|
|
7 |
Application : Image Recognition, Image Classification, Medical Imaging |
|
|
8 |
</pre> |
|
|
9 |
|
|
|
10 |
### Description |
|
|
11 |
<pre> |
|
|
12 |
1. Detected Pneumonia from Chest X-Ray images using Custom Deep Convololutional Neural Network and by retraining pretrained model “InceptionV3” with 5856 images of X-ray (1.15GB). |
|
|
13 |
2. For retraining removed output layers, freezed first few layers and fine-tuned model for two new label classes (Pneumonia and Normal). |
|
|
14 |
3. With Custom Deep Convololutional Neural Network attained testing accuracy 89.53% and loss 0.41. |
|
|
15 |
</pre> |
|
|
16 |
|
|
|
17 |
|
|
|
18 |
|
|
|
19 |
<b>Model Parameters</b> |
|
|
20 |
Machine Learning Library: Keras |
|
|
21 |
Base Model : InceptionV3 && Custom Deep Convolutional Neural Network |
|
|
22 |
Optimizers : Adam |
|
|
23 |
Loss Function : categorical_crossentropy |
|
|
24 |
|
|
|
25 |
<b>For Custom Deep Convolutional Neural Network : </b> |
|
|
26 |
<b>Training Parameters</b> |
|
|
27 |
Batch Size : 64 |
|
|
28 |
Number of Epochs : 30 |
|
|
29 |
Training Time : 2 Hours |
|
|
30 |
|
|
|
31 |
<b>Output (Prediction/ Recognition / Classification Metrics)</b> |
|
|
32 |
<b>Testing</b> |
|
|
33 |
Accuracy (F-1) Score : 89.53% |
|
|
34 |
Loss : 0.41 |
|
|
35 |
Precision : 88.37% |
|
|
36 |
Recall (Pneumonia) : 95.48% (For positive class) |
|
|
37 |
<!--Specificity : --> |
|
|
38 |
</pre> |
|
|
39 |
|
|
|
40 |
|
|
|
41 |
<kbd> |
|
|
42 |
<a href="[https://i.imgur.com/km4MF3J.png](https://imgur.com/km4MF3J)"></a> |
|
|
43 |
</kbd> |
|
|
44 |
|
|
|
45 |
|
|
|
46 |
|
|
|
47 |
#### Tools / Libraries |
|
|
48 |
<pre> |
|
|
49 |
Languages : Python |
|
|
50 |
Tools/IDE : Google Colab |
|
|
51 |
Libraries : Keras, TensorFlow, Inception, ImageNet |
|
|
52 |
</pre> |
|
|
53 |
|