一种大米图像分割模型构建方法

本发明属于粮食质量检测领域，具体涉及一种大米图像分割模型构建方法。该方法对大米图像分割模型进行训练构建；模型内步骤包括：1)将首个特征层作为当前层；对输入图像特征提取得到相邻不同尺度的特征图作为当前层输入；2)通过当前层FPM模块得到不同尺度的增强图，与上一特征层Swin Transformer输出经过当前层第一融合模块后再与上一特征层Swin Transformer的输出融合，输入当前层的Swin Transformer，之后与当前层的增强图经过当前层第二融合模块后再与当前层对应尺度的增强图融合得到当前层的输出；若不存在下一特征层则根据当前层输出与当前层Swin Transformer输出的融合结果完成实例分割；否则将当前层输出作为下一特征层的输入；将下一特征层作为新的当前层，重复2)直至完成实例分割。