forked from dydx-git/COVID-Classifier
-
Notifications
You must be signed in to change notification settings - Fork 0
/
readme.txt
35 lines (23 loc) · 1.82 KB
/
readme.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
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.