-
公开(公告)号:CN120070460A
公开(公告)日:2025-05-30
申请号:CN202510118334.7
申请日:2025-01-24
Applicant: 北京工业大学
IPC: G06T7/10 , G06N3/0464 , G06N3/0455 , G06N3/08
Abstract: 本发明公开了一种基于注意力机制的口腔影像分割方法及系统,方法包括:构建由编码器块、跳跃连接模块以及解码器块组成的分割模型,编码器块包括下采样卷积和通道注意力模块,解码器块包括上采样卷积和非局部模块,跳跃连接模块融合编码器块和解码器块提取的特征;训练模型时,基于交叉熵损失函数、Dice损失函数调整模型参数,当模型性能不再提升或达到预设迭代次数,停止训练,得到训练后的模型;建待分割数据集,将其输入至分割模型,待分割数据集经编码器块卷积后得到深层特征图,通道注意力模块对深层特征图进行处理得到关键特征图,解码器块非局部模块对关键特征图进行处理得到全局特征图,基于关键特征图、全局特征图输出口腔分割影像。