一种基于U-Net的多模块融合的大豆豆荚识别方法

本发明提供一种基于U‑Net的多模块融合的大豆豆荚识别方法，包括以下步骤：构建U‑Net网络；获取晃动状态下拍摄的大豆豆荚模糊图像，并将其送入U‑Net网络；通过下采样模块对大豆豆荚图像进行四次下采样，并得到最小特征图；将最小特征图送入IDAM模块并获得注意力特征图；通过上采样模块对注意力特征图进行四次上采样，获得大豆豆荚清晰图像。在u‑net骨干网络进行上采样之前加入IDAM模块，增强网络捕捉大豆豆荚图像之间相关性的能力，提高对豆荚区域的关注度；通过引入EFEM模块对大豆豆荚图像边缘进行映射以得到更加精准的豆荚区域的边缘信息，提高了模型的Dice和Recall，从而得到大头豆荚的清晰图。