APP下载

基于RAU-net的视网膜血管图像分割

2021-12-23张丽娟李超然

红外技术 2021年12期

张丽娟,梅 畅,李超然,章 润

基于RAU-net的视网膜血管图像分割

张丽娟1,梅 畅1,李超然2,章 润1

(1. 长春工业大学 计算机科学与工程学院,吉林 长春 130012;2. 长春理工大学 光电信息学院信息工程学院,吉林 长春 130114)

在眼科疾病的诊断中,对视网膜血管进行分割是非常有效的一种方法。在方法使用中,经常会遇到由于视网膜血管背景对比度低及血管末梢细节复杂导致的血管分割难度较大的问题,通过在设计网络的过程中在基础U-net网络中引入残差学习,注意力机制等模块,并将两者巧妙地结合在一起,提出一种新型的基于U-net的RAU-net视网膜血管图像分割算法。首先,在网络的编码器阶段加入残差模块,解决了模型网络加深导致梯度爆炸以及梯度消失的问题。其次,在网络的解码器阶段引入注意力门(attention gate, AU)模块,用来抑制不必要的特征,从而使模型产生更高的精度。通过在DRIVE数据集上进行验证,该算法的准确率、灵敏度、特异性和1-score分别达到了0.7832,0.9815,0.9568和0.8192。分割效果相对于普通监督学习算法较为良好。

图像分割;视网膜血管;全卷积网络;残差模块;注意力机制

0 引言

人眼的视网膜结构的表征和分析可以帮助准确诊断各种眼部疾病[1]以及各种其他疾病,例如青光眼,糖尿病性视网膜病变(diabetic retinopathy, DR)[2],年龄相关性黄斑变性(age-related macular degeneration, AMD)等。为此,对视网膜血管树结构的分析和准确检测可以被认为是大规模人群预防疾病的首要任务。然而,由于视网膜图像的噪声影响,图像对比度过低,微小血管之间的变化不均匀,对视网膜图像的分割成为了许多专家研究的问题。……

登录APP查看全文