-
题名基于改进U-Net的糖尿病视网膜渗出物分割
- 1
-
-
作者
程小辉
李贺军
邓昀
陶小梅
黎辛晓
-
机构
桂林理工大学信息科学与工程学院
桂林理工大学广西嵌入式技术与智能系统重点实验室
-
出处
《计算机工程与设计》
北大核心
2023年第11期3489-3495,共7页
-
基金
国家自然科学基金项目(61906051)
广西自然科学基金项目(2018GXNSFAA281235)
广西中青年教师基础能力提升基金项目(2018KY0248、2020KY06026)。
-
文摘
为解决当前深度分割模型对糖尿病视网膜病变渗出物区域精准分割的问题,提出一种基于改进U-Net的糖尿病视网膜渗出物分割方法。重构编码端的特征提取器,以ResNet50预训练模型作为特征提取器,提升模型对低维特征的学习能力;结合注意力机制优化解码端上采样过程,缓解卷积特征丢失的问题,提高模型对病灶特征通道权重的学习能力;在模型解码端构造多特征尺度的融合,丰富不同尺度特征的语义信息。利用IDRID数据集和DIARETDB1数据集进行训练,训练采用Focal loss焦点损失函数,模型的特异性、灵敏度、精准率和F1-Score分别达到0.973、0.982、0.971和0.875,Dice系数达到了0.984,与其它先进算法相比,该算法分割性能较优。
-
关键词
糖尿病视网膜
编码端
解码端
渗出物分割
特征提取
注意力机制
焦点损失
-
Keywords
diabetic retinopathy
encoder
decoder
exudate segmentation
feature extraction
attention mechanism
focal loss
-
分类号
TP391.41
[自动化与计算机技术—计算机应用技术]
TP183
[自动化与计算机技术—控制理论与控制工程]
-