基于改进YOLOv8的小麦不完善粒双面判别和重量估算方法

本发明公开了基于改进YOLOv8的小麦不完善粒双面判别和重量估算方法，涉及小麦品质检测技术领域；该方法包括以下步骤：构建双面数据集：分别采集若干小麦籽粒的正面图像和背面图像，将该正面图像和背面图像处理得到双面数据集；构建改进YOLOv8模型：基于YOLOv8网络模型融合注意力模块，且对Concat连接结构替换成BiFPN连接结构，得到改进YOLOv8模型；训练改进YOLOv8模型；小麦籽粒的双面判别：改进YOLOv8模型对待识别若干小麦籽粒的正面图像和背面图像进行分割，整合双面分割结果，得到类别判别结果；小麦籽粒的重量估计。本发明适合小麦不完善粒细粒度特征分割，并能进一步进行籽粒重量的估算，为小麦质检流程提供自动化服务。