摘要
本发明公开了一种基于掩码自注意力机制的脑肿瘤MRI图像分割方法,所述分割方法包括:S1:构建一个3D Masked Swin Transformer模块;S2:构建一个三维医学图像分割模型;S3:获取原始医学图像数据集,对原始医学图像数据集进行数据预处理,并将预处理后的医学图像数据集划分为训练集和测试集;S4:设置网络模型训练的超参数和在线数据增强方法,使用所述训练集训练三维医学图像分割模型;S5:使用训练完成的三维医学图像分割模型对原始脑部MRI医学图像中的肿瘤区域进行分割。本发明的优点在于:此方法能够有效地将Transformer应用至医学图像分割中,泛化能力强,抗干扰能力和分割准确率高。
- 单位