ZHANG Lijuan, MEI Chang, LI Chaoran, ZHANG Run. Retinal Vessel Image Segmentation Based on RAU-net[J]. Infrared Technology , 2021, 43(12): 1222-1227,1233.
Citation: ZHANG Lijuan, MEI Chang, LI Chaoran, ZHANG Run. Retinal Vessel Image Segmentation Based on RAU-net[J]. Infrared Technology , 2021, 43(12): 1222-1227,1233.

Retinal Vessel Image Segmentation Based on RAU-net

More Information
  • Received Date: October 07, 2020
  • Revised Date: October 13, 2020
  • In the diagnosis of ophthalmic diseases, segmentation of retinal blood vessels is a quite effective method. However, using this method, difficulties in blood vessel segmentation are often encountered due to the low contrast of the retinal blood vessel background and complex details of the blood vessel end. Thus, residual learning is introduced by adding a basic U-net network to the process of network design. Through the introduction of residual learning and attention mechanism modules into the basic U-net network in the process of network design, a new type of U-net-based RAU-net retinal blood vessel image segmentation algorithm is proposed. First, the residual module is added to the encoder stage of the network to address gradient explosion and disappearance caused by the deepening of the model network. Second, the attention gate module is introduced in the decoder stage of the network to suppress unnecessary features to ensure high accuracy of the model. Through verification of DRIVE, the accuracy, sensitivity, specificity, and F1-score of the algorithm reached 0.7832, 0.9815, 0.9568, and 0.8192, respectively. Thus, the segmentation effect was better than that of ordinary supervised learning algorithms
  • [1]
    Abràmoff M D, Folk J C, Han D P, et al. Automated analysis of retinal images for detection of referable diabetic retinopathy[J]. JAMA Ophthalmol, 2013, 131(3): 351-357. DOI: 10.1001/jamaophthalmol.2013.1743
    [2]
    ZENG X, CHEN H, LUO Y, et al. Automated diabetic retinopathy detection based on binocular siamese-like convolutional neural network[J]. IEEE Access, 2019, 7: 30744-30753 (doi: 10.1109/ACCESS.2019.2903171).
    [3]
    Chaudhuri S, Chatterjee S, Katz N, et al. Detection of blood vessels in retinal images using two-dimensional matched filters[J]. IEEE Transactions on Medical Imaging, 1989, 8(3): 263-269. DOI: 10.1109/42.34715
    [4]
    Vlachos M, Dermatas E. Multi-scale retinal vessel segmentation using line tracking[J]. Computerized Medical Imaging & Graphics the Official Journal of the Computerized Medical Imaging Society, 2010, 34(3): 213-227. http://www.onacademic.com/detail/journal_1000034065429410_d1d5.html
    [5]
    Zana F, Klein J C. Segmentation of vessel-like patterns using mathematical morphology and curvature evaluation[J]. IEEE Transactions on Image Processing, 2001, 10(7): 1010-1019. DOI: 10.1109/83.931095
    [6]
    WU C, YI B, ZHANG Y, et al. Retinal vessel image segmentation based on improved convolutional neural network[J]. Acta Optica Sinica, 2018, 38(11): 1111004-1-1111004-7. DOI: 10.3788/AOS201838.1111004
    [7]
    Dharmawan D A, LI D, Ng B P, et al. A new hybrid algorithm for retinal vessels segmentation on fundus images[J]. IEEE Access, 2019, 7: 41885-41896 (doi: 10.1109/ACCESS.2019.2906344).
    [8]
    FU H, XU Y, WONG D, et al. Retinal vessel segmentation via deep learning network and fully-connected conditional random fields[C]//13th International Symposium on Biomedical Imaging of IEEE (ISBI 2016), 2016: 698-701.
    [9]
    JIANG Y, TAN N, PENG T, et al. Retinal vessels segmentation based on dilated multi-scale convolutional neural network[J]. IEEE Access, 2019, 7: 76342-76352. DOI: 10.1109/ACCESS.2019.2922365
    [10]
    Alom M Z, Yakopcic C, Taha T M, et al. Nuclei segmentation with recurrent residual convolutional neural networks based U-Net (R2U-Net)[C]//IEEE National Aerospace and Electronics Conference (NAECON 2018), 2018: 228-233(doi: 10.1109/NAECON.2018.8556686).
    [11]
    ZHUANG J. Multi-path networks based on U-Net for medical image segmentation[J]. arXiv preprint, 2018, arXiv: 1810.07810.
    [12]
    HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2016: 770-778.
    [13]
    Oktay O, Schlemper J, Folgoc L L, et al. Attention U-net: Learning where to look for the pancreas[J]. arXiv preprint, 2018, arXiv: 1804.03999.
    [14]
    Gkantsidis C, Mihail M, Saberi A. Random walks in peer-to-peer networks: algorithms and evaluation[J]. Performance Evaluation, 2006, 63(3): 241-263. DOI: 10.1016/j.peva.2005.01.002
    [15]
    Staal J, Abramoff M D, Niemeijer M, et al. Ridge-based vessel segmentation in color images of the retina[J]. IEEE Transactions on Medical Imaging, 2004, 23(4): 501-509. DOI: 10.1109/TMI.2004.825627
    [16]
    Satapathy S C, Raja N S M, et al. Multi-level image thresholding using Otsu and chaotic bat algorithm[J]. Neural Computing and Applications, 2018, 29(12): 1285-1307. DOI: 10.1007/s00521-016-2645-5
    [17]
    PAN X, ZHANG Q, ZHANG H, et al. A fundus retinal vessels segmentation scheme based on the improved deep learning U-Net model[J]. IEEE Access, 2019, 7(99): 122634-122643. http://ieeexplore.ieee.org/document/8796382
    [18]
    雷军明, 王成钢, 刘雪燕. 基于改进U-Net的视网膜血管图像分割[J]. 信息技术与信息化, 2019(10): 168-169. DOI: 10.3969/j.issn.1672-9528.2019.10.052

    LEI J M, WANG C G, LIU X Y. Retinal vessel image segmentation based on improved U-Net[J]. Information Technology and Information Technology, 2019(10): 168-169. DOI: 10.3969/j.issn.1672-9528.2019.10.052
    [19]
    Kingma D, Ba J. Adam: A method for stochastic optimization[J]. arXiv preprint, 2014, arXiv: 1412.6980.
  • Related Articles

    [1]ZHANG Huining, ZHANG Peihang. Infrared Small Target Detection Based on Improving Attention Mechanism[J]. Infrared Technology , 2025, 47(6): 748-756.
    [2]YE Ye. A Deep Learning Method for Hyperspectral Detection of Heavy Metal Contaminants in Soil Based on Attention Mechanism[J]. Infrared Technology , 2025, 47(4): 453-458.
    [3]ZHAO Yating, HAN Long, HE Huihuang, CHEN Chu. DSEL-CNN: Image Fusion Algorithm Combining Attention Mechanism and Balanced Loss[J]. Infrared Technology , 2025, 47(3): 358-366.
    [4]QI Yanjie, HOU Qinhe. Infrared and Visible Image Fusion Combining Multi-scale and Convolutional Attention[J]. Infrared Technology , 2024, 46(9): 1060-1069.
    [5]WANG Yan, ZHANG Jinfeng, WANG Likang, FAN Xianghui. Underwater Image Enhancement Based on Attention Mechanism and Feature Reconstruction[J]. Infrared Technology , 2024, 46(9): 1006-1014.
    [6]CHONG Fating, DONG Zhangyu, YANG Xuezhi, ZENG Qingwang. SAR and Multispectral Image Fusion Based on Dual-channel Multi-scale Feature Extraction and Attention[J]. Infrared Technology , 2024, 46(1): 61-73.
    [7]LI Xiangrong, SUN Lihui. Multiscale Infrared Target Detection Based on Attention Mechanism[J]. Infrared Technology , 2023, 45(7): 746-754.
    [8]CHEN Xin. Infrared and Visible Image Fusion Using Double Attention Generative Adversarial Networks[J]. Infrared Technology , 2023, 45(6): 639-648.
    [9]LUO Di, WANG Congqing, ZHOU Yongjun. A Visible and Infrared Image Fusion Method based on Generative Adversarial Networks and Attention Mechanism[J]. Infrared Technology , 2021, 43(6): 566-574.
    [10]WANG Hao, ZHANG Jingjing, LI Yuanyuan, WANG Feng, XUN Lina. Hyperspectral Image Classification Based on 3D Convolution Joint Attention Mechanism[J]. Infrared Technology , 2020, 42(3): 264-271.

Catalog

    Article views (152) PDF downloads (41) Cited by()
    Related

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return