Optical Instruments, Volume. 46, Issue 5, 9(2024)

Retinal blood vessel segmentation algorithm based on improved U-Net

Yuan LIU, Baicheng LI*, and Chunbo WU
Author Affiliations
  • School of Optical-Electrical and Computer Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China
  • show less

    The existing algorithms have the problem of low segmentation accuracy when facing small vessels in retinal images. In this paper, an improved U-Net segmentation algorithm is proposed by introducing residual module and detail enhancement attention mechanism module into U-Net network. In the coding and decoding stages, the residual module is used to replace the traditional convolutional module, which solves the problem of network degradation with increasing depth. Meanwhile, a detail enhancement attention mechanism is added between the encoder and the decoder to reduce the useless information in the output of the encoder, so that the sensitivity of the network to capture valid feature information is improved. In addition, the experimental results based on the standard image set DRIVE reveal that the segmentation accuracy, sensitivity and F1 score of the proposed algorithm are improved by 0.46%, 2.14% and 1.56%, respectively compared to the U-Net, which is superior to the traditional segmentation algorithms.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yuan LIU, Baicheng LI, Chunbo WU. Retinal blood vessel segmentation algorithm based on improved U-Net[J]. Optical Instruments, 2024, 46(5): 9

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Aug. 28, 2023

    Accepted: --

    Published Online: Jan. 3, 2025

    The Author Email: LI Baicheng (lbcusst@163.com)

    DOI:10.3969/j.issn.1005-5630.202308280111

    Topics