[发明专利]基于注意力及边界约束的息肉图像分割方法、装置、设备及介质

说明书

本发明实施例公开了一种基于注意力及边界约束的息肉图像分割方法、装置、设备及介质。本发明涉及图像处理技术领域，其包括：获取待分割息肉图像及Ground Truth图像；将待分割息肉图像进行特征提取，得到多个低层特征及多个高层特征；将扩大感受野及上采样后的多个高层特征叠加得到全局映射图；根据第二低层特征及Ground Truth图像生成边缘映射图，并通过第一预设损失函数对边缘映射图进行监督以更新第二低层特征；对多个高层特征、全局映射图以及更新后的第二低层特征进行平衡注意力学习得到多个预测图，并根据第二预设损失函数对多个预测图及Ground Truth图像进行监督约束，得到息肉分割图。本申请实施例可提高息肉图像的分割准确率。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种基于注意力及边界约束的息肉图像分割方法、装置、设备及介质。

背景技术

结直肠癌包括结肠癌和直肠癌，是一种高发的恶性肿瘤，而大多数结直肠癌是由结肠或直肠上的息肉演变而来，因此，息肉的早期筛查对结直肠癌的预防至关重要。现有技术中，内窥镜检查因利于医生直观地观察病灶的形态特征，被认为是诊断结息肉的金标准，但内窥镜检查高度依赖经验丰富的临床医生，临床医师诊断一个病人的待分割息肉图像需要花费较长时间，极大地降低了息肉的筛查效率，而且息肉通常呈现出各种形状、大小和纹理特征，甚至边界不清晰且难定位，极大地降低了息肉图像分割的准确率。

发明内容

本发明实施例提供了一种基于注意力及边界约束的息肉图像分割方法、装置、设备及介质，旨在解决现有息肉图像分割准确率不高的问题。

第一方面，本发明实施例提供了一种基于注意力及边界约束的息肉图像分割方法，应用于构建的息肉分割模型，其包括：

获取待分割息肉图像及与所述待分割息肉图像相对应的Ground Truth图像；

将所述待分割息肉图像进行特征提取，得到多个低层特征及多个高层特征，其中，所述多个低层特征包括第一低层特征及由所述第一低层特征卷积后的第二低层特征；

对所述多个高层特征扩大感受野得到多个目标特征图，并将上采样后的所述多个目标特征图进行特征叠加得到全局映射图；

根据所述第二低层特征及所述Ground Truth图像生成边缘映射图，并通过第一预设损失函数对所述边缘映射图进行监督以更新所述第二低层特征；

根据所述第二预设损失函数及所述Ground Truth图像对所述多个高层特征、所述全局映射图以及更新后的所述第二低层特征进行平衡注意力学习得到多个预测图，并根据第二预设损失函数对所述多个预测图及所述Ground Truth图像进行监督约束，得到息肉分割图。

第二方面，本发明实施例还提供了一种基于注意力及边界约束的息肉图像分割装置，应用于构建的息肉分割模型，其包括：

获取处理单元，用于获取待分割息肉图像及与所述待分割息肉图像相对应的Ground Truth图像；

特征提取单元，用于将所述待分割息肉图像进行特征提取，得到多个低层特征及多个高层特征，其中，所述多个低层特征包括第一低层特征及由所述第一低层特征卷积后的第二低层特征；

特征聚合单元，用于对所述多个高层特征扩大感受野得到多个目标特征图，并将上采样后的所述多个目标特征图进行特征叠加得到全局映射图；

边界约束单元，用于根据所述第二低层特征及所述Ground Truth图像生成边缘映射图，并通过第一预设损失函数对所述边缘映射图进行监督以更新所述第二低层特征；

平衡注意力单元，用于对所述多个高层特征、所述全局映射图以及更新后的所述第二低层特征进行平衡注意力学习得到多个预测图，并根据第二预设损失函数对所述多个预测图及所述Ground Truth图像进行监督约束，得到息肉分割图。