计算机与现代化Issue(5)：110-114,5.DOI:10.3969/j.issn.1006-2475.2024.05.019
基于注意力机制的U-Net眼底图像分割算法
An Attention Mechanism-based U-Net Fundus Image Segmentation Algorithm
摘要
Abstract
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.关键词
视网膜眼底图像分割/注意力机制/DropOut层Key words
retinal fundus image segmentation/attention mechanism/DropOut layer分类
信息技术与安全科学引用本文复制引用
张子旭,李嘉莹,栾鹏鹏,彭圆圆..基于注意力机制的U-Net眼底图像分割算法[J].计算机与现代化,2024,(5):110-114,5.基金项目
江西省大学生创新创业项目(202310404008) （202310404008）
