基于通道注意力和多头注意力机制的医学图像分割方法

    公开(公告)号:CN118537350A

    公开(公告)日:2024-08-23

    申请号:CN202410491641.5

    申请日:2024-04-23

    IPC分类号: G06T7/11 G06N3/0464

    摘要: 本发明公开了基于通道注意力和多头注意力机制的医学图像分割方法,涉及医学图像分割技术领域。本发明至少包括以下步骤:S1:数据预处理,为了更好地利用数据集图片之间的关系并适应网络的输入,经过预处理之后得到了2D切片;S2:特征提取;S3:特征恢复,采用编码器将提取后的特征进行恢复。本发明在编码器中使用通道注意力和卷积多头注意力混合机制进行特征提取,能够弥补各自的缺陷,同时捕获通道特征和全局信息,且为了抑制来自编码器的无关信息,有效地利用高低层特征,并引入了带有BiFusion融合模块的级联解码器,通过实验结果表明,所提出的方法在ACDC和Synapse数据集上的表现优于最先进的方法,突出了未来潜在的研究方向,以提高医学图像分割的准确率。