Huang Hua, Shang Zhenhong, Yu Chunhui
Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China.
Yunnan Key Laboratory of Artificial Intelligence, Kunming University of Science and Technology, Kunming 650500, China.
Biomed Opt Express. 2024 Apr 26;15(5):3344-3365. doi: 10.1364/BOE.522482. eCollection 2024 May 1.
Accurate and automated retinal vessel segmentation is essential for performing diagnosis and surgical planning of retinal diseases. However, conventional U-shaped networks often suffer from segmentation errors when dealing with fine and low-contrast blood vessels due to the loss of continuous resolution in the encoding stage and the inability to recover the lost information in the decoding stage. To address this issue, this paper introduces an effective full-resolution retinal vessel segmentation network, namely FRD-Net, which consists of two core components: the backbone network and the multi-scale feature fusion module (MFFM). The backbone network achieves horizontal and vertical expansion through the interaction mechanism of multi-resolution dilated convolutions while preserving the complete image resolution. In the backbone network, the effective application of dilated convolutions with varying dilation rates, coupled with the utilization of dilated residual modules for integrating multi-scale feature maps from adjacent stages, facilitates continuous learning of multi-scale features to enhance high-level contextual information. Moreover, MFFM further enhances segmentation by fusing deeper multi-scale features with the original image, facilitating edge detail recovery for accurate vessel segmentation. In tests on multiple classical datasets,compared to state-of-the-art segmentation algorithms, FRD-Net achieves superior performance and generalization with fewer model parameters.
准确且自动化的视网膜血管分割对于视网膜疾病的诊断和手术规划至关重要。然而,传统的U形网络在处理细小且对比度低的血管时,由于在编码阶段分辨率连续丢失以及在解码阶段无法恢复丢失的信息,常常会出现分割错误。为了解决这个问题,本文引入了一种有效的全分辨率视网膜血管分割网络,即FRD-Net,它由两个核心组件组成:主干网络和多尺度特征融合模块(MFFM)。主干网络通过多分辨率扩张卷积的交互机制实现水平和垂直扩展,同时保持完整的图像分辨率。在主干网络中,有效应用具有不同扩张率的扩张卷积,并利用扩张残差模块整合相邻阶段的多尺度特征图,有助于持续学习多尺度特征以增强高级上下文信息。此外,MFFM通过将更深层次的多尺度特征与原始图像融合,进一步增强分割效果,有助于恢复边缘细节以实现准确的血管分割。在多个经典数据集上的测试中,与最先进的分割算法相比,FRD-Net以更少模型参数实现了卓越的性能和泛化能力。