An Attention Mechanism-based U-Net Fundus Image Segmentation Algorithm
The radius and width of retinal fundus vessels are important indicators for assessing eye diseases,so accurate segmen-tation of fundus images is becoming increasingly meaningful.In order to effectively assist doctors in diagnosing eye diseases,the paper proposes a new neural network to segment fundus vascular images.The basic idea is to reduce the information loss by im-proving the traditional U-Net model with the help of an attention fusion mechanism,using Transformer to construct a channel at-tention mechanism and a spatial attention mechanism,and fusing the information obtained by the two attention mechanisms.In addition,the number of retinal fundus images is relatively small,and the coefficients of the neural network are relatively large,which are prone to overfitting during training,so the DropBlock layer is introduced to solve this problem.The proposed method was validated on the publicly available dataset DRIVE and compared with several state-of-the-art methods.The results show that our method achieved the highest ACC value of 0.967 and the highest F1 value of 0.787.These experimental results demon-strate that the proposed method is effective in segmenting retinal fundus images.
国家科技期刊平台
NSTL
万方数据
