Acta Optica Sinica, Volume. 40, Issue 10, 1010001(2020)

Improved U-Net Segmentation Algorithm for the Retinal Blood Vessel Images

Daxiang Li1,2 and Zhen Zhang1、*
Author Affiliations
  • 1College of Communication and Infornation Technology, Xi'an University of Posts and Telecommunications, Xi'an, Shaanxi 710121, China
  • 2Key Laboratory of Ministry of Public Security, Electronic Information Field Inspection and Application Technology, Xi'an, Shaanxi 710121, China
  • show less

    In this study, we propose an improved U-Net retinal vascular image segmentation algorithm by introducing some modules, such as inception, hole convolution, and attention mechanism, into the U-Net network to solve the problem of low segmentation accuracy caused by the small blood vessels in the retinal image. Initially, the inception module was added during the encoding stage, and convolution kernels of different scales were used to extract the image features to obtain multiscale information from the image. Subsequently, a cascaded hole convolution module was added to the bottom of the U-Net network for expanding the receptive field of the convolution operation without increasing the network parameters. Finally, an attention mechanism was designed for the deconvolution operation during the decoding phase. The problem of weight dispersion can be solved by focusing on the target features under the combination of the attention mechanism and jump connection mode. The experimental results obtained using the standard image set DRIVE denote that the average accuracy, sensitivity, and specificity of the proposed algorithm are 1.15%, 6.15%, and 0.67% higher than those of the traditional U-Net algorithm, respectively, and that the proposed algorithm outperforms other traditional segmentation algorithms.

    Tools

    Get Citation

    Copy Citation Text

    Daxiang Li, Zhen Zhang. Improved U-Net Segmentation Algorithm for the Retinal Blood Vessel Images[J]. Acta Optica Sinica, 2020, 40(10): 1010001

    Download Citation

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

    Category: Image Processing

    Received: Jan. 8, 2020

    Accepted: Feb. 26, 2020

    Published Online: Apr. 28, 2020

    The Author Email: Zhang Zhen (zhang408356262@163.com)

    DOI:10.3788/AOS202040.1010001

    Topics