利索能及
我要发布
收藏
专利号: 2022110898133
申请人: 闽江学院
专利类型:发明专利
专利状态:已下证
更新日期:2026-10-22
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,包括:

构建视网膜血管图像分割网络模型;所述视网膜血管图像分割网络模型通过多尺度特征卷积块,使用不同大小的卷积核来进行特征提取,利用不同特征尺度信息,对血管整体和末端细微血管进行有效提取分割,并使用非局部注意力模块,获取更加丰富的全局语义信息;而后在跳跃连接部分采用多尺度注意力门控网络,输入不同维度的特征图,有选择性的学习相互关联的区域;

采用训练数据集对所述视网膜血管图像分割网络模型进行训练,得到训练好的视网膜血管图像分割网络模型;

通过训练好的视网膜血管图像分割网络模型对视网膜血管图像进行图像分割。

2.根据权利要求1所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,在所述视网膜血管图像分割网络模型中,编码器的每一步都包括一个多尺度特征卷积块和一个2×2最大池化操作,在每个下采样中,将特征通道的数量加倍;解码器中的每个步都包括一个用于上采样的2×2转置卷积操作,并将特征通道的数量减半,与来自编码器的相应特征映射级联;在编码器和解码器之间设置非局部注意力模块。

3.根据权利要求1所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,所述多尺度特征卷积块使用1×1、3×3、5×5三种不同大小的卷积核来提取多尺度特征进行融合,以提高微血管特征提取能力;卷积结果通过concatenate的方式融合为一个组合特征图,再经过一个1×1的卷积,保持通道数与输入特征图一致;然后使用DropBlock有效的丢弃多尺度特征图中部分语义信息,防止网络过拟合;输入的特征图通过

1×1的卷积添加到经过DropBlock的多尺度特征图中,形成残差结构;结构块中加入ReLU激活函数,以避免过拟合。

4.根据权利要求1所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,所述非局部注意力模块通过计算特征中每个位置的加权和,来增加全局特征之间的关联性,在较远距离相关联的特征像素点之间建立依赖联系,以更好的捕获全局的语义信息;所述非局部注意力模块对输入特征图进行线性映射分别生成三个特征图M1、M2、M3,通过reshape和transpose得到M1、M2的乘法矩阵,并通过softmax函数得到全局的空间注意图A;将空间注意力图A应用在M3上并通过transpose和线性卷积得到最终的非局部注意力特征图;在输出端加入残差连接,与输入特征图相加。

5.根据权利要求4所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,所述非局部注意力模块的计算公式为:

Zi=Aiyi+xi     (2)

其中,i与j为输出位置,f函数用于计算i与j位置的相似度;g函数用于计算特征图在j点的表示;响应函数C(x)用于对累加后的结果进行归一化处理;yi为生成的非局部特征图;

Zi为经过残差网络得到的最终的空间注意力特征图。

6.根据权利要求1所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,所述多尺度注意力门控网络用于突出血管部分的特征相应以及抑制图像背景区域的特征响应,所述多尺度注意力门控网络包括多个多尺度注意力门模块,所述多尺度注意力门模块首先将编码特征图x、浅层编码特征图y以及要进行上采样的特征图z的通道数和尺寸调整一致,然后将x、z分别经过最大池化、ReLU激活函数、一个1×1卷积和BN归一化,将y经过一个1×1卷积;令x与y相加,经过ReLU激活函数以及一个1×1卷积和Sigmoid激活操作,再与y相乘,得到的结果与z相加,经过ReLU激活函数以及一个1×1卷积和Sigmoid激活操作,再与y相乘;如果y是第一级编码特征图,仅进行y与z的特征图操作,然后通过concatenate的方式进行自身融合再经过ReLU激活函数以及一个1×1卷积和BN归一化,得到输出特征图;由于浅层和深层的特征图有不同的类别信息和位置信息,故浅层特征图配合深层特征图可得到丰富的语义信息,减缓无用信息的干扰;将多尺度注意力门模块加到神经网络的跳跃连接部分,以对视网膜特征进行空间增强,减少噪声对视网膜分割的影响。

7.根据权利要求1所述的基于多尺度注意力门控网络的视网膜血管图像分割方法,其特征在于,对所述视网膜血管图像分割网络模型进行训练时,使用的损失函数为交叉熵损失函数,其定义如下:式中,i表示类别种类;yi表示真实值的概率;pi表示模型预测的概率值。