一种基于正交注意力的粮食籽粒图像分割方法及系统

本发明属于图像分割领域，具体涉及一种基于正交注意力的粮食籽粒图像分割方法及系统。该方法包括：将粮食籽粒图像输入训练好的分割模型得到结果；分割模型为特征提取模块中包含正交注意力单元的Unet架构；正交注意力单元对输入特征图沿高度轴旋转，经增强得到宽度方向权重再与沿高度轴旋转后的特征图相乘并旋转还原，得到宽度的增强图；对输入特征图沿宽度轴旋转，经增强得到高度方向权重再与沿宽度轴旋转后的特征图相乘并旋转还原，得到高度的增强图；对输入特征图恒等映射，经增强得到通道注意权重，再与恒等映射后的特征图相乘并二次恒等映射得到通道的增强图；将宽度、高度和通道的增强图融合；增强包括压缩、激活变换与激活函数依次处理操作。