Dataset Open Access

Robust Chest CT Image Segmentation of COVID-19 Lung Infection based on limited data

Dominik Müller; Iñaki Soto Rey; Frank Kramer

The coronavirus disease 2019 (COVID-19) affects billions of
lives around the world and has a significant impact on public
healthcare. Due to rising skepticism towards the sensitivity of
RT-PCR as screening method, medical imaging like computed
tomography offers great potential as alternative. For this
reason, automated image segmentation is highly desired as
clinical decision support for quantitative assessment and
disease monitoring. However, publicly available COVID-19
imaging data is limited which leads to overfitting of traditional
approaches. To address this problem, we propose an innovative
automated segmentation pipeline for COVID-19 infected
regions, which is able to handle small datasets by utilization as
variant databases. Our method focuses on on-the-fly
generation of unique and random image patches for training
by performing several preprocessing methods and exploiting
extensive data augmentation. For further reduction of the
overfitting risk, we implemented a standard 3D U-Net
architecture instead of new or computational complex neural
network architectures. Through a 5-fold cross-validation on 20
CT scans of COVID-19 patients, we were able to develop a
highly accurate as well as robust segmentation model for lungs
and COVID-19 infected regions without overfitting on the
limited data. Our method achieved Dice similarity coefficients
of 0.956 for lungs and 0.761 for infection. We demonstrated
that the proposed method outperforms related approaches,
advances the state-of-the-art for COVID-19 segmentation and
improves medical image analysis with limited data. The code
and model are available under the following link:
https://github.com/frankkramer-lab/covid19.MIScnn

Code: https://github.com/frankkramer-lab/covid19.MIScnn
Files (2.7 GB)
Name Size
evaluation.spe.zip
md5:a91ecfc9972194a30438853579dc3165
9.2 MB Download
evaluation.zip
md5:e586bd041f563bbd18d0ec0130ac2fc9
158.7 MB Download
models.zip
md5:20ac6f5186d7e98edbd54bc547d50ad6
2.5 GB Download
predictions.zip
md5:3fb40f3f9c05f224ec22421a7b9ad450
17.6 MB Download
1,560
407
views
downloads
All versions This version
Views 1,560242
Downloads 40745
Data volume 575.3 GB34.7 GB
Unique views 1,348215
Unique downloads 18122

Share

Cite as