README.md 3.21 KB
Newer Older
lindawangg's avatar
lindawangg committed
1 2 3 4 5 6
# COVID-Net and COVIDx Dataset
The COVID-19 pandemic continues to have a devastating effect on the health and well-being of global population. A critical step in the fight against COVID-19 is effective screening of infected patients, with one of the key screening approaches being radiological imaging using chest radiography. It was found in early studies that patients present abnormalities in chest radiography images that are characteristic of those infected with COVID-19.  Motivated by this, a number of artificial intelligence (AI) systems based on deep learning have been proposed and results have been shown to be quite promising in terms of accuracy in detecting patients infected with COVID-19 using chest radiography images. However, to the best of the authors' knowledge, these developed AI systems have been closed source and unavailable to the research community for deeper understanding and extension, and unavailable for public access and use. Therefore, in this study we introduce COVID-Net, a deep convolutional neural network design tailored for the detection of COVID-19 cases from chest radiography images that is open source and available to the general public. We also describe the chest radiography dataset leveraged to train COVID-Net, which we will refer to as COVIDx and is comprised of 5941 posteroanterior chest radiography images across 2839 patient cases from two open access data repositories. Furthermore, we investigate how COVID-Net makes predictions using an explainability method in an attempt to gain deeper insights into critical factors associated with COVID cases, which can aid clinicians in improved screening. By no means a production-ready solution, the hope is that the open access COVID-Net, along with the description on constructing the open source COVIDx dataset, will be leveraged and build upon by both researchers and citizen data scientists alike to accelerate the development of highly accuracy yet practical deep learning solutions for detecting COVID-19 cases and accelerate treatment of those who need it the most.

If you would like to contribute COVID-19 x-ray images, please contact us at linda.wang513@gmail.com and a28wong@uwaterloo.ca/alex@darwinai.ca. Lets all work together to stop the spread of COVID-19!

## Training
lindawangg's avatar
lindawangg committed
7 8 9
To train:
* download npy files from [here](https://drive.google.com/file/d/1zCnmcMxSRZTqJywur7jCqZk0z__Mevxp/view?usp=sharing)
* use train.ipynb to train model
lindawangg's avatar
readme  
lindawangg committed
10 11 12

## Chest X-Ray and CT Dataset
Combined datasets from https://github.com/ieee8023/covid-chestxray-dataset and https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia
lindawangg's avatar
stats  
lindawangg committed
13

lindawangg's avatar
lindawangg committed
14
Combined dataset can be downloaded [here](https://drive.google.com/file/d/1-T26bHP7MCwB8vWeKufjGmPKl8pesM1J/view?usp=sharing)
lindawangg's avatar
readme  
lindawangg committed
15

lindawangg's avatar
stats  
lindawangg committed
16 17 18 19 20
Stats
|  Type | Normal | Viral | Bacteria | COVID-19 |
|:-----:|:------:|:-----:|:--------:|:--------:|
| train |  1349  |  1355 |   2540   |    60    |
|  test |   234  |  149  |    246   |     8    |
Linda Wang's avatar
Linda Wang committed
21 22

## License
lindawangg's avatar
lindawangg committed
23
Our desire is to encourage broad adoption and contribution to this project. Accordingly this project has been licensed under the GNU Affero General Public License 3.0. If you would like to discuss alternative licensing models, please reach out to us at: linda.wang513@gmail.com and a28wong@uwaterloo.ca or alex@darwinai.ca