Skip to content

An efficient machine learning model to assist in the diagnosis of COVID-19 infection in chest x-ray images

Notifications You must be signed in to change notification settings

abzargar/COVID-Classifier

Repository files navigation

Abstract:

Chest-X ray (CXR) radiography can be used as a first-line triage process for non-COVID-19 patients who have similar symptoms. 
However, the similarity between features of CXR images of COVID-19 and pneumonia caused by other infections make the differential 
diagnosis by radiologists challenging. We hypothesized that machine learning-based classifiers can reliably distinguish the CXR images 
of COVID-19 patients from other forms of pneumonia. We used feature extraction and dimensionality reduction methods to generate an efficient 
machine learning classifier that can distinguish COVID-19 cases from non-COVID-19 cases with high accuracy and sensitivity. We propose that 
our COVID-Classifier classifier can be used in conjunction with other tests for optimal allocation of hospital resources by rapid triage of non-COVID-19 cases.  

Dataset:
COVID-> 140 X-ray images
Normal-> 140 X-ray images
Pneumonia-> 140 X-ray images

How to use:
1-Run "preprocess_images.py" to preprocess images done by resizing, normalization, adaptive histogram equalization

2-Run "extract_features.py" to create three feature pools for covid or normal or pneumonia datasets

3-Run "evaluate_features.py" to evaluate extracted features

4-Run "train_model.py" to train and then evaluate model  

Test results:

	        Precision	 Sensitivity	 F-score	 Support
COVOD-19	   96%	           100%	           0.98	           25
Normal	           88%	           100%	           0.94	           31
Pneumonia	   100%	           82%	           0.91	           28

Please cite the follwoing paper if you use our paper codes:

Abolfazl Zargari Khuzani, Morteza Heidari, Ali Shariati, "COVID-Classifier: An automated machine learning model to assist in the diagnosis of COVID-19 infection in chest x-ray images," medRxiv, doi: https://doi.org/10.1101/2020.05.09.20096560, 2020.

About

An efficient machine learning model to assist in the diagnosis of COVID-19 infection in chest x-ray images

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages