Pretrained (imagenet) ResNet50 with 3 classification. Achieves 97.2% test accuracy. Use Grad-CAM for infected area segmentation.
Gamma transform and CLAHE
| COVID-19 | Viral Pneumonia | Normal | |
|---|---|---|---|
| Total | 756 | 1361 | 1422 |
| Train | 600 | 1200 | 1200 |
| Test | 156 | 161 | 222 |