The global outbreak of COVID-19 has become an important research topic in healthcare since 2019. RT-PCR is the main method for detecting COVID-19, but the long detection time is a problem. Therefore, the pathological study of COVID-19 with CT image is an important supplement to RT-RCT. The current TVLoss-based segmentation promotes the connectivity of diseased areas. However, normal pixels between some adjacent diseased areas are wrongly identified as diseased pixels. In addition, the proportion of diseased pixels in CT images is small, and the traditional BCE-based U-shaped network only focuses on the whole CT without diseased pixels, which leads to blurry border and low contrast in the predicted result. In this way, this paper proposes a SCTV-UNet to solve these problems. By combining spatial and channel attentions on the encoder, more visual layer information are obtained to recognize the normal pixels between adjacent diseased areas. By using the composite function DTVLoss that focuses on the pixels in the diseased area, the problem of blurry boundary and low contrast caused by the use of BCE in traditional U-shaped networks is solved. The experiment shows that the segmentation effect of the proposed SCTV-UNet has significantly improved by comparing with the SOTA COVID-19 segmentation networks, and can play an important role in the detection and research of clinical COVID-19.
Keywords: COVID-19; Channel attention; DTVLoss; SCTV-UNet; Spatial attention.
© The Author(s), under exclusive licence to Springer-Verlag GmbH Germany, part of Springer Nature 2023, Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.