- The dataset is named COV19-CT-DB. It is obtained at AI-enabled Medical Image Analysis Workshop and Covid-19 Diagnosis Competition (ICCV 2021 Workshop: MIA-COV19D 2021). The project is conducted using chest CT-scan series of images aiming to develop an automated solution for Covid-19/non-Covid-19 diagnosis.
- The team (IDU-CVLab) introduces a light weight solution for COVID-19 diagnosis and is listed on the leaderboard here.
The algorithm introduces a less hand-engineered CNN model Architecture for automated COVID-19 diagnosis.
The CNN model achitechture is in picture below:
* The work makes use of the above-mentioned CNN model, with slices preprocessing before training.
Slices preprocessing -as in here- includes a static rectangular cropping, and non-representative slcies removal in each CT scan. For more details, please refer to the peer-reviewed paper.
Gussian noise was added to the images in the original validation set and then images are processed as mentioned above and the pretrained and saved CNN model was tested on the newly-created noisey images. This step aims to check our method's performance against noisey data to prove the solution's robustness. Salt-and-paper was also added to the images to further validate the results with different types of noise. The python code is named "Noisey-Images-Image-Processing-Light-CNN-Model.py".
To replicate the codes, the following must be noted:
- To run the code properly you would need a training set of images and a validation set of images.
- The images must be put in the appropriate directories. With that, the directory of training and validation images included in the code should be changed to match the directory where your image datasets are located. This method is following the documentation for ‘imagedatagenerator’ and ‘flow_from_directory’ at https://www.tensorflow.org/api_docs/python/tf/keras/preprocessing/image/ImageDataGenerator
- Please note: this is a binary classificaiton task. To replicate the code using muti-class classificaiton data, you need to modify the model's output to suit your task.
▪ numpy == 1.19.5
▪ matplotlib == 3.3.4
▪ tensorflow == 2.5.0
▪ CV2 == 4.5.4
▪ sklearn == 0.24.2
If you find this method useful, please consider citing:
@article{doi:10.1080/21681163.2023.2219765,
author = {Kenan Morani and D. Unay},
title = {Deep learning-based automated COVID-19 classification from computed tomography images},
journal = {Computer Methods in Biomechanics and Biomedical Engineering: Imaging & Visualization},
volume = {0},
number = {0},
pages = {1-16},
year = {2023},
publisher = {Taylor & Francis},
doi = {10.1080/21681163.2023.2219765},