科研之友
微信
新浪微博
Facebook
分享链接
摘要
为解决当前深度分割模型对糖尿病视网膜病变渗出物区域精准分割的问题，提出一种基于改进U-Net的糖尿病视网膜渗出物分割方法。重构编码端的特征提取器，以ResNet50预训练模型作为特征提取器，提升模型对低维特征的学习能力；结合注意力机制优化解码端上采样过程，缓解卷积特征丢失的问题，提高模型对病灶特征通道权重的学习能力；在模型解码端构造多特征尺度的融合，丰富不同尺度特征的语义信息。利用IDRID数据集和DIARETDB1数据集进行训练，训练采用Focal loss焦点损失函数，模型的特异性、灵敏度、精准率和F1-Score分别达到0.973、0.982、0.971和0.875,Dice系数达到了0.984,与其它先进算法相比，该算法分割性能较优。
- 单位
