本发明公开了一种基于改进U‑Net网络的不同容重小麦籽粒图像分割算法，采用残差堆叠模块作为下采样过程中的主要骨干结构，在特征融合部分，在初级特征图后通过嵌入CBAM注意力机制模块，从通道和空间位置以自适应地调整不同像素点的特征融合权重；在解码器部分中嵌入自注意力模块，增强不同目标之间的相关性。本发明实现了对不同容重的小麦籽粒从图像角度进行识别检测，改变了传统的采用容重筒进行小麦容重检测的方法，从而可以提升效率，为后续应用图像方法实现小麦容重的检测奠定了坚实的基础。