Li Angran, Sun Mingzhu, Wang Zengshuo
College of Artificial Intelligence, Nankai University, Tianjin 300350, China.
Bioengineering (Basel). 2024 May 14;11(5):488. doi: 10.3390/bioengineering11050488.
Retinal vessel segmentation plays a crucial role in medical image analysis, aiding ophthalmologists in disease diagnosis, monitoring, and treatment guidance. However, due to the complex boundary structure and rich texture features in retinal blood vessel images, existing methods have challenges in the accurate segmentation of blood vessel boundaries. In this study, we propose the texture-driven Swin-UNet with enhanced boundary-wise perception. Firstly, we designed a Cross-level Texture Complementary Module (CTCM) to fuse feature maps at different scales during the encoding stage, thereby recovering detailed features lost in the downsampling process. Additionally, we introduced a Pixel-wise Texture Swin Block (PT Swin Block) to improve the model's ability to localize vessel boundary and contour information. Finally, we introduced an improved Hausdorff distance loss function to further enhance the accuracy of vessel boundary segmentation. The proposed method was evaluated on the DRIVE and CHASEDB1 datasets, and the experimental results demonstrate that our model obtained superior performance in terms of Accuracy (ACC), Sensitivity (SE), Specificity (SP), and F1 score (F1), and the accuracy of vessel boundary segmentation was significantly improved.
视网膜血管分割在医学图像分析中起着至关重要的作用,有助于眼科医生进行疾病诊断、监测和治疗指导。然而,由于视网膜血管图像中复杂的边界结构和丰富的纹理特征,现有方法在准确分割血管边界方面存在挑战。在本研究中,我们提出了具有增强边界感知能力的纹理驱动Swin-UNet。首先,我们设计了一个跨层纹理互补模块(CTCM),在编码阶段融合不同尺度的特征图,从而恢复在降采样过程中丢失的细节特征。此外,我们引入了一个逐像素纹理Swin块(PT Swin块)来提高模型定位血管边界和轮廓信息的能力。最后,我们引入了一种改进的豪斯多夫距离损失函数,以进一步提高血管边界分割的准确性。所提出的方法在DRIVE和CHASEDB1数据集上进行了评估,实验结果表明,我们的模型在准确率(ACC)、灵敏度(SE)、特异性(SP)和F1分数(F1)方面取得了优异的性能,并且血管边界分割的准确性得到了显著提高。