一种基于注意力机制的息肉图像分割方法
摘要:
本发明公开一种基于注意力机制的息肉图像分割网络,其特点是采用级联部分解码器聚合息肉高级特征生成息肉全局特征图和息肉分割结果图,同时基于注意力机制构造息肉语义增强模块,重新整合通道维度与空间维度的息肉语义信息,增强有效语义信息的流动,进而增强退化的息肉语义信息。本发明具体步骤包括:数据集的预处理、基于注意力机制的息肉图像分割网络模型的构建、基于注意力机制的息肉图像分割网络模型的训练与测试。相较于现有技术,本发明同时注意到息肉特征提取过程中小息肉的位置语义和边缘模糊息肉的边界语义退化的问题,提高了网络模型对于小目标息肉和边缘模糊息肉的分割能力,实现息肉图像的自动分割,具有良好的应用前景。
0/0