一种基于反向注意力网络的图像分割方法技术

本发明专利技术公开了一种基于反向注意力网络的图像分割方法，首先获取图像数据集，构建训练集和测试集；然后构建反向注意力网络模型，反向注意力网络模型的处理过程包括图像经过多个卷积层逐层编码依次得到不同层次的输出特征层，利用不同层次的输出特并行拼接聚合后输入到解码器进行解码得到全局特征图像，将全局特征图像与输出特征输入到反向注意网络中进行处理，直到得到低层次的反向注意特征；将训练集输入反向注意力网络模型进行训练，获得训练好的反向注意力网络模型，获取图像分割结果。本发明专利技术利用图像编码得到的高级特征输入到反向注意网络中使得图像分割的精度大幅提高。反向注意网络中使得图像分割的精度大幅提高。

【技术实现步骤摘要】
一种基于反向注意力网络的图像分割方法


[0001]本专利技术属于医学图像处理领域，具体涉及一种基于反向注意网络的图像分割方法。

技术介绍

[0002]结直肠癌(CRC)是世界上第三种最常见的癌症。因此，通过结直肠肿瘤前检查预防结直肠肿瘤已成为世界范围内的一项非常重要的健康检查。结肠镜检查可以提供结直肠息肉的位置和外观信息，使医生能够在结肠息肉发展为结直肠癌之前将其切除，是一种有效的结直肠癌筛查和预防技术。许多研究表明，早期结肠镜检查有助于降低30％的CRC发病率。因此，在临床上，精确的息肉分割是非常重要的。然而，由于息肉通常在外观上有所不同，例如大小、颜色和质地，即使它们是同一类型的。
[0003]在现有结肠镜图像分割方法中，息肉与其周围黏膜之间的边界通常是模糊的，并且缺乏分割方法所需的强烈对比度。这些问题导致了息肉的不精确分割，有时甚至会导致息肉的漏检。因此，一种能够在早期发现所有可能的息肉的自动、准确的息肉分割方法对预防大肠癌具有重要意义。

技术实现思路

[0004]为了解决现有技术的上述问题，本专利技术提供一种基于反向注意网络的图像分割方法，利用图像编码得到的高级特征输入到反向注意网络中使得图像分割的精度大幅提高。本专利技术的技术方案如下：
[0005]一种基于反向注意力网络的图像分割方法，包括以下步骤：
[0006]S1，获取图像数据集，构建训练集和测试集；
[0007]S2，构建反向注意力网络模型，其中反向注意力网络模型的处理过程具体为：
[0008]训练集图像经过多个卷积层逐层编码依次得到不同层次的输出特征层 f1、f2、f3、f4和f5，根据不同层次的输出特征f3、f4和f5并行拼接聚合后输入到解码器进行解码得到全局特征图像Sg；
[0009]将全局特征图像Sg进行一次下采样操作后，与输出特征f5输入到第一反向注意网络中进行处理，得到输出特征R5；第一反向注意网络处理的过程包括对全局特征图像Sg进行感兴趣区域反转操作，并将反转后的图像与输出特征f5进行相乘，得到输出特征R5；将输出特征R5和全局特征图像Sg进行融合得到反向注意特征S5；
[0010]将反向注意特征S5进行一次上采样操作后，与输出特征f4输入到第二反向注意网络中进行处理，得到输出特征R4；第二反向注意网络处理的过程包括对反向注意特征S5进行感兴趣区域反转操作，并将反转后的图像与输出特征f4进行相乘，得到输出特征R4；将输出特征R4和反向注意特征S5进行融合得到反向注意特征S4；
[0011]将反向注意特征S4进行一次上采样操作后，与输出特征f3输入到第三反向注意网络中进行处理，得到输出特征R3；第三反向注意网络处理的过程包括对反向注意特征S4进
行感兴趣区域反转操作，并将反转后的图像与输出特征f3进行相乘，得到输出特征R3；将输出特征R3和反向注意特征S4进行融合得到反向注意特征S3；
[0012]反向注意特征S3通过Sigmoid函数激活后得到反向注意力网络模型的处理结果；
[0013]S3，将训练集输入反向注意力网络模型进行训练，获得训练好的反向注意力网络模型；
[0014]S4，将测试集输入训练好的反向注意力网络模型，获取图像分割结果。
[0015]进一步地，感兴趣区域为息肉区域。
[0016]本专利技术的有益效果在于：能够利用图像编码得到的高级特征输入到反向注意网络中使得图像分割的精度大幅提高。
附图说明
[0017]图1是本专利技术基于反向注意力网络的图像分割方法结构示意图；
[0018]图2是本专利技术基于反向注意力网络的图像分割方法的流程图；
[0019]图3是本专利技术反向注意力网络的结构示意图；
[0020]图4是本专利技术中感兴趣区域反转操作前的示意图；
[0021]图5是本专利技术中感兴趣区域反转操作后的示意图；
[0022]附图标记：RA1为第一反向注意网络、RA2为第二反向注意网络、RA3为第一反向注意网络。
具体实施方式
[0023]下面结合附图和实施例，对本专利技术技术方案作进一步描述：
[0024]本实施例中提供了一种基于反向注意网络的图像分割方法，如图2所示，包括：
[0025]步骤一，获取图像数据集，构建训练集和测试集。
[0026]在本申请实施例中，从标准数据集Kvasir中获取图像数据集，并将获取的图像数据集中的80％的图像数据作为训练集，20％的图像数据作为测试集，并将输入调整为352
×
352。
[0027]步骤二，构建反向注意力网络模型，其中反向注意力网络模型的处理过程具体为：
[0028]反向注意力网络模型具体是首先粗略地定位息肉区域，然后根据局部特征准确地提取其轮廓模板。
[0029]在本实施例中如图1，在步骤1中得到的息肉训练集图像经过5个卷积层逐层编码依次得到不同层次的输出特征层f1、f2、f3、f4和f5，其中，f1 和f2为低级层次特征，f3、f4和f5为高级层次特征，根据不同层次的输出特征f3、f4和f5并行拼接聚合后输入到解码器进行解码得到全局特征图像Sg；
[0030]在本申请实施例中具体通过公式(1)得到反向注意网络生成的输出特征：
[0031]R
i
＝f
i
×
A
i
ꢀꢀ
(1)
[0032]其中，i＝3,4,5，f
i
为高级层次特征f3、f4和f5，A
i
为反向注意权重。
[0033]具体的反向注意权重通过：
[0034]A
i
＝Θ(σ(p(S
i+1
)))
ꢀꢀ
(2)
[0035]其中，Θ表示矩阵E减去输入的逆运算，矩阵E的所有元素都是1，σ表示Sigmoid函
数，p表示上采样操作，S表示反向注意特征。
[0036]具体的，在本实施例中，将全局特征图像Sg进行一次下采样操作后，与输出特征f5输入到第一反向注意网络中进行处理，得到输出特征R5；第一反向注意网络处理的过程包括对全局特征图像Sg进行感兴趣区域反转操作，如图4，原来息肉区域(白色区域)的注意为0.9，非息肉区域(黑色区域)的注意为0.1，在进行感兴趣区域反转操作之后，如图5，得到息肉区域(白色区域)的注意为0.1，非息肉区域(黑色区域)为0.9.并将反转后的图像与输出特征f5进行相乘，得到输出特征R5；将输出特征R5和全局特征图像Sg 进行融合得到反向注意特征S5；
[0037]将反向注意特征S5进行一次上采样操作后，与输出特征f4输入到第二反向注意网络中进行处理，得到输出特征R4；第二反向注意网络处理的过程包括对反向注意特征S5进行感兴趣区域反转操作，并将反转后的图像与输出特征f4进行相乘，得到输出特征R4；将输出特征R4和反向注意特征S5进行融合得到反向注意特征S4；
[003本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于反向注意力网络的图像分割方法，其特征在于，包括以下步骤：S1，获取图像数据集，构建训练集和测试集；S2，构建反向注意力网络模型，其中所述反向注意力网络模型的处理过程具体为：训练集图像经过多个卷积层逐层编码依次得到不同层次的输出特征层f1、f2、f3、f4和f5，根据所述不同层次的输出特征f3、f4和f5并行拼接聚合后输入到解码器进行解码得到全局特征图像Sg；将所述全局特征图像Sg进行一次下采样操作后，与输出特征f5输入到第一反向注意网络中进行处理，得到输出特征R5；所述第一反向注意网络处理的过程包括对所述全局特征图像Sg进行感兴趣区域反转操作，并将反转后的图像与所述输出特征f5进行相乘，得到输出特征R5；将所述输出特征R5和所述全局特征图像Sg进行融合得到反向注意特征S5；将所述反向注意特征S5进行一次上采样操作后，与输出特征f4输入到第二反向注意网络中进行处理，得到输出特征R4；所述第二反向注意网络处...

【专利技术属性】
技术研发人员：王博，赵威，申建虎，张伟，徐正清，
申请(专利权)人：西安智诊智能科技有限公司，
类型：发明
国别省市：