/covid19_detection_in_pytorch

ResNet-18 model for image classification of COVID-19 chest x-ray images in PyTorch.

Primary LanguageJupyter Notebook

covid19_detection_in_pytorch

ResNet-18 model for image classification of COVID-19 chest x-ray images in PyTorch.

Custom Dataset and Dataloader.

Open In Collab

Before training:

alt text

Final result:

alt text

Dataset references

[1]https://bimcv.cipf.es/bimcv-projects/bimcv-covid19/#1590858128006-9e640421-6711

[2]https://github.com/ml-workgroup/covid-19-image-repository/tree/master/png

[3]https://sirm.org/category/senza-categoria/covid-19/

[4]https://eurorad.org

[5]https://github.com/ieee8023/covid-chestxray-dataset

[6]https://figshare.com/articles/COVID-19_Chest_X-Ray_Image_Repository/12580328

[7]https://github.com/armiro/COVID-CXNet

[8]https://www.kaggle.com/c/rsna-pneumonia-detection-challenge/data

[9] https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia