基于改进YOLOX的小麦不完善粒检测技术研究

    Research on unsound kernels detection technology based on improved YOLOX

    • 摘要: 针对现有小麦不完善粒检测方法存在适用性差、识别效率低的问题,提出一种基于改进YOLOX的小麦不完善粒检测算法。在3个小麦品种、不同拍摄高度和不同籽粒数量场景下采集小麦图像数据集。针对籽粒过小难以检测的问题,引入坐标注意力模型提升图像中不完善粒的显著度,并使用加权双向特征金字塔改进特征融合模块结构,实现了不同尺度不完善粒特征的有效融合。针对多目标检测实时性问题,使用深度可分离卷积模块轻量化特征提取网络,降低网络的计算参数量,提高检测速度。结果表明,改进后的模型检测漏检率小于5%,对不完善粒的平均检测精度达到93.43%,检测速度为37 fps,相较于原YOLOX网络,检测精度和速度分别提高了2.58百分点和5.78 fps。该方法可以在不同籽粒大小、颗粒数量以及不同密集程度情况下对小麦不完善粒进行有效识别检测,可为进一步的小麦不完善粒实时检测和统计应用提供技术参考。

       

      Abstract: This study was aimed to solve the problems of poor applicability and low recognition efficiency of existing unsound wheat kernels detection methods, an improved YOLOX-based wheat imperfection grain detection method was proposed. Wheat images of three varieties, different heights and number of seeds scenarios were collected to produce a dataset. Meanwhile, this study introduced coordinate attention model in YOLOX backbone network to enhance the saliency of imperfect grains in images, and the feature fusion structure is improved by using a weighted bidirectional feature pyramid to realize the effective fusion of imperfect grain features at different scales. To address the problem of real-time multi-target detection, the depth-separable convolution module was used to lighten the feature extraction network, reduce the computational power of the network, and improve the detection speed of the network. The results showed that the improved model had a detection rate of less than 5%, an average detection accuracy of 93.43%, and a detection speed of 37 fps, which were 2.58% and 5.78 fps higher than those of the original YOLOX network. The improved model can effectively identify and detect wheat imperfect grains at different number of seeds and different densities, which can provide technical reference for further real-time detection and statistical application of wheat imperfect grains.

       

    /

    返回文章
    返回