张雷,张良,刘刚,等.基于改进UNet的视网膜血管分割算法[J]. 微电子学与计算机,2023,40(2):101-109. doi: 10.19304/J.ISSN1000-7180.2022.0341
引用本文: 张雷,张良,刘刚,等.基于改进UNet的视网膜血管分割算法[J]. 微电子学与计算机,2023,40(2):101-109. doi: 10.19304/J.ISSN1000-7180.2022.0341
ZHANG L,ZHANG L,LIU G,et al. Retinal vascular segmentation algorithm based on improved UNet[J]. Microelectronics & Computer,2023,40(2):101-109. doi: 10.19304/J.ISSN1000-7180.2022.0341
Citation: ZHANG L,ZHANG L,LIU G,et al. Retinal vascular segmentation algorithm based on improved UNet[J]. Microelectronics & Computer,2023,40(2):101-109. doi: 10.19304/J.ISSN1000-7180.2022.0341

基于改进UNet的视网膜血管分割算法

Retinal vascular segmentation algorithm based on improved UNet

  • 摘要: 针对视网膜血管结构复杂、图像对比度低、细节区域分割不精准问题,提出一种基于改进UNet的分割算法. 首先,结合Diverse Branch Block多分支、多尺度思想,在编解码路径上增加DBB-ConvNet模块,该模块组合了不同尺度、不同复杂度的分支来丰富特征空间的多样性,进而提升网络的特征学习和表达能力;其次,为加强特征重用,避免冗余特征影响,在网络底端加入Dense-net密集连接;最后,为进一步提升分割效果,在跳跃连接用BConvLSTM结合非线性函数处理编解码路径间的特征映射,替换原始的简单串联. 该算法在公开数据集DRIVE和CHASE_DB1上的实验结果表明,对比其他算法,本文所提出的算法在视觉和各项客观评价指标上均有较好的分割效果. 与U-Net算法相比,在两种数据集上本文所提算法在精确率、F1值上分别提升了1.97%、2.04%与2.05%、5.86%.

     

    Abstract: Aiming at the problems of complex retinal vascular structure, low image contrast and inaccurate detail region segmentation, a segmentation algorithm based on improved UNet is proposed. Firstly, combined with the idea of multi branch and multi-scale of divide branch block, a DBB-ConvNet module is added to the coding and decoding path. The module combines branches with different scales and complexity to enrich the diversity of feature space, so as to improve the ability of feature learning and expression of the network; Secondly, in order to strengthen feature reuse and avoid the influence of redundant features, dense connection is added at the bottom of the network; Finally, in order to further improve the segmentation effect, bconvlstm combined with nonlinear function is used to deal with the feature mapping between code and decoding paths in jump connection, replacing the original simple concatenation. Based on open datasets DRIVE and CHASE_DB1, The experimental results show that compared with other algorithms, the proposed algorithm has better segmentation effect in vision and various objective evaluation indexes. Compared with U-Net algorithm, the accuracy and F1 value of the proposed algorithm are improved by 1.97%, 2.04% and 2.05%, 5.86% respectively.

     

/

返回文章
返回