Diagnosis of COVID-19 infection using three-dimensional semantic segmentation and classification of computed tomography images

Abstract

Coronavirus 19 (COVID-19) can cause severe pneumonia that may be fatal. Correct diagnosis is essential. Computed tomography (CT) usefully detects symptoms of COVID-19 infection. In this retrospective study, we present an improved framework for detection of COVID-19 infection on CT images; the steps include pre-processing, segmentation, feature extraction/ fusion/selection, and classification. In the pre-processing phase, a Gabor wavelet filter is applied to enhance image intensities. A marker-based, watershed controlled approach with thresholding is used to isolate the lung region. In the segmentation phase,COVID-19 lesions are segmented using an encoder- /decoder-based deep learning model in which deepLabv3 serves as the bottleneck and mobilenetv2 as the classification head. DeepLabv3 is an effective decoder that helps to refine segmentation of lesion boundaries. The model was trained using fine-tuned hyperparameters selected after extensive experimentation. Subsequently, the Gray Level Co-occurrence Matrix (GLCM) features and statistical features including circularity, area, and perimeters were computed for each segmented image. The computed features were serially fused and the best features (those that were optimally discriminatory) selected using a Genetic Algorithm (GA) for classification. The performance of the method was evaluated using two benchmark datasets: The COVID-19 Segmentation and the POF Hospital datasets. The results were better than those of existing methods.