Skip to content

pvti/SIIM-COVID-19-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Visitors

👩‍⚕️ Identification and Localization COVID-19 Abnormalities on Chest Radiographs

1Université de Toulon, Aix Marseille Université, CNRS, LIS, UMR 7020, FranceCorresponding Author
Solutions to screen and diagnose positive patients for SARS-CoV-2 promptly and efficiently are critical in the context of the COVID-19 pandemic’s complex evolution. Recent researches have demonstrated the efficiency of deep learning and particularly convolutional neural networks (CNNs) in classifying and detecting lung disease-related lesions from radiographs. This paper presents a solution using ensemble learning techniques on advanced CNNs to classify as well as localize COVID-19-related abnormalities in radiographs. Two classifiers including EfficientNetV2 and NFNet are combined with three detectors, DETR, Yolov7, and EfficientDet. Along with gathering and training the model on a large number of datasets, image augmentation, and cross-validation are also addressed. Since then, this study has shown promising results and has received excellent marks in the Society for Imaging Informatics in Medicine’s competition. The analysis in model selection for the trade-off between speed and accuracy is also given.

📋 News

  • [2023.01.02] Paper is accepted to AICV 2023.
  • [2022.09.06] Create baseline.

🦠 Main results

Evaluation of the COVID-19 lesion detector on the SIIM test set:

Detector Accuracy (mAP@ IoU 0.5:0.95) Performance
Discrete model Lung localized Inference speed (FPS) GPU memory requirement (MB) Model size (MB)
DETR 0.542 0.587 25 2683 232
Yolov7 0.563 0.591 34 3520 290
EfficientDet 0.499 0.574 19 1903 187
Weighted Box Fusion 0.605 0.612 8 8106 709

💉 Installation

Please refer to INSTALL.md for installation instructions.

🧬 Model zoo

Trained models are available in the MODEL_ZOO.md.

💻 Dataset zoo

Please see DATASET_ZOO.md for detailed description of the training/evaluation datasets.

🔍 Getting Started

Follow the aforementioned instructions to install environments and download models and datasets.

GETTING_STARTED.md provides a brief intro of the usage of builtin command-line tools.

🔬 Citing

If you use this work in your research or wish to refer to the results, please use the following BibTeX entry.

@inproceedings{pham2023identification,
  title={Identification and localization COVID-19 abnormalities on chest radiographs},
  author={Pham, Van Tien and Nguyen, Thanh Phuong},
  booktitle={The International Conference on Artificial Intelligence and Computer Vision},
  pages={251--261},
  year={2023},
  organization={Springer}
}