基于EfficientNet-B5级联U型网络的视网膜血管分割方法 |
| |
作者姓名: | 雷帮军 陈玮华 夏平 张光一 |
| |
作者单位: | 1. 三峡大学水电工程智能视觉监测湖北省重点实验室;2. 三峡大学计算机与信息学院 |
| |
基金项目: | 国家自然科学基金(U1401252);;湖北省重点实验室开放基金(2018SDSJ07)项目资助; |
| |
摘 要: | 针对视网膜图像细小血管难以分割以及病灶误分割等问题,提出了一种基于EfficientNet-B5级联U型网络的视网膜血管分割方法。首先,采用EfficientNet-B5编码器对图像特征编码,提取图像更深层特征;其次,引入空洞空间卷积池化金字塔模块有效地提取视网膜图像中的多尺度特征信息;在双卷积块中加入高效通道注意力捕获跨通道交互的信息;然后,采用级联U型结构,后一级网络对前一级网络的血管分割结果进行细化与优化;最后,采用形态学算子对网络输出的分割图像进行后处理。在CHASE_DB1数据集上的实验结果表明,该算法的F1值、特异度、准确率及AUC分别为79.57%、98.21%、96.30%和97.51%,相比FC-RCF、U-Net和Res-UNet等算法均有提升。该算法在精确地分割血管的同时表现出较好的抗干扰性能。
|
关 键 词: | 视网膜血管分割 级联U型网络 EfficientNet-B5编码器 高效通道注意力 |
|