摘要
本发明提供了一种基于真菌生长优化算法的自注意力机制,属于医学图像的分割技术领域。解决了现有UNet网络在医学图像分割任务中存在的模型参数量大、计算复杂度高、对不规则病灶边缘分割精度不足的技术问题。其技术方案为:包括以下步骤:步骤一、对输入的医学图像进行预处理;步骤二、使用真菌生长优化算法的自注意力机制来替代Transfomer中的自注意力机制;步骤三、将预处理好的图像数据集输入到编码器和解码器网络中进行训练,得到最优模型;步骤四、训练完成后,将医学图像验证集输入到已经得到的最优模型中。本发明能够更精确地分割病灶,提高了准确率和鲁棒性。
技术关键词
真菌
前馈神经网络
算法
编码器解码器
加载医学图像数据
医学图像分割方法
医学图像数据集
动态多通道
通道注意力机制
分支
初始化方法