Skip to content

Latest commit

 

History

History
245 lines (167 loc) · 6.28 KB

README.md

File metadata and controls

245 lines (167 loc) · 6.28 KB

COVID19-Detection

This projects involves experimentation with XRAY dataset to detect COVID19. I used transfer learning and performed different experiments to improve the accuracy of the model. I also used Focal Loss Function to solve the problem of imbalance dataset.

DataSet Can be Found Here COVID Datasets2.

Weights can be Found Here Model Weights.

VGG16 with Binary Cross Entropy

Training Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 174 26
Not COVID19 1 5999
Class Pneumonia Not Pneumonia
Pneumonia 2023 177
Not Pneumonia 7 3993
Class Normal Not Normal
Normal 3995 5
Not Normal 180 2020

Validation Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 17 11
Not COVID19 0 600
Class Pneumonia Not Pneumonia
Pneumonia 202 26
Not Pneumonia 2 398
Class Normal Not Normal
Normal 399 1
Not Normal 28 200

ResNet18 with Binary Cross Entropy

Training Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 188 12
Not COVID19 7 5993
Class Pneumonia Not Pneumonia
Pneumonia 1971 229
Not Pneumonia 7 3993
Class Normal Not Normal
Normal 3993 7
Not Normal 221 1979

Validation Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 22 6
Not COVID19 4 596
Class Pneumonia Not Pneumonia
Pneumonia 213 15
Not Pneumonia 9 391
Class Normal Not Normal
Normal 391 9
Not Normal 15 213

VGG16 with Focal Loss

Training Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 190 10
Not COVID19 8 5992
Class Pneumonia Not Pneumonia
Pneumonia 2133 67
Not Pneumonia 80 3920
Class Normal Not Normal
Normal 3913 87
Not Normal 61 2139

Validation Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 23 5
Not COVID19 0 600
Class Pneumonia Not Pneumonia
Pneumonia 212 16
Not Pneumonia 10 390
Class Normal Not Normal
Normal 391 9
Not Normal 16 212

ResNet18 with Focal Loss

Training Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 192 8
Not COVID19 55 5945
Class Pneumonia Not Pneumonia
Pneumonia 2119 81
Not Pneumonia 242 3758
Class Normal Not Normal
Normal 3760 240
Not Normal 87 2113

Validation Data

COVID19 Class Pneumonia Class Normal Class
Class COVID19 Not COVID19
COVID19 25 3
Not COVID19 4 596
Class Pneumonia Not Pneumonia
Pneumonia 215 13
Not Pneumonia 29 371
Class Normal Not Normal
Normal 368 32
Not Normal 13 215

Experiments Details:

Model Loss Functions Focal Loss Parms F1 Score: COVID19
VGG16 BCE 0.81
ResNet18 BCE 0.76
VGG16 Focal Loss a =2, g =2 0.90
ResNet18 Focal Los a1 = 1 g=2 0.88