基于YOLO v5-MDC的重度粘连小麦籽粒检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2019YFD1002401)


Detection Method of Severe Adhesive Wheat Grain Based on YOLO v5-MDC Model
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    小麦籽粒检测在千粒质量计算及作物育种方面有着重要应用,重度粘连籽粒的有效检测是其关键。本研究设计了一种YOLO v5-MDC的轻量型网络用于重度粘连小麦籽粒检测。该网络在YOLO v5s检测网络的基础上,用混合深度可分离卷积(Mixed depthwise convolutional, MDC)模块进行改进,同时将MDC模块与压缩激励(Squeeze and excitation, SE)模块相结合,以达到在基本不损失模型精度的前提下减少模型参数的目的。YOLO v5-MDC网络将YOLO v5s特征提取网络骨干部分的卷积、归一化、激活函数(Convolution, Batch normal, Hardswish, CBH)模块替换为MDC模块,减少了模型的参数,经过500次迭代训练,模型的精确率P为93.15%,召回率R为99.96%,平均精度均值(mAP)为99.46%。根据模型在测试集上的检测效果,本研究探究了训练次数、不同光源与不同拍摄距离对模型检测结果的影响,统计结果表明,在绿色光源下模型检测精确率最高,为98.00%,在5cm拍摄高度下图像的检测精确率最高,为98.60%。同时本研究在50次迭代下与YOLO v5s、RetinaNet、YOLO v4网络模型的检测效果进行了对比,结果表明,YOLO v5-MDC的mAP为99.40%,比YOLO v5s模型降低了0.06个百分点,但模型所占存储空间最小,仅为13.4MB,比YOLO v5s模型减少了0.6MB,对于单幅图像的最大检测时间为0.08s,平均检测时间为0.03s。综上,本研究所设计模型能有效实现重度粘连小麦籽粒的检测,同时模型检测效率高,所占存储小,可为小麦籽粒检测嵌入式设备研发提供技术支持。

    Abstract:

    Wheat grain detection has important applications in the calculation of thousand grain weight and crop breeding, and the effective detection of heavily adhesive grains is the key issue should be solved. A lightweight network called YOLO v5-MDC was designed for the detection of heavily adhesive wheat grains to provide technical support for the development of mobile terminals. The YOLO v5s detection network was chosen and the mixed depthwise convolutional (MDC) module was carried out to improve it. At the same time, the MDC module combined with a squeeze and excitation(SE) module was applied to achieve the purpose of reducing model parameters without losing the accuracy of the model. The YOLO v5-MDC network replaced the convolution, batch normal, Hardswish (CBH) modules of the backbone of the YOLO v5s feature extraction network with the MDC module, reducing the model parameters. After 500 iterations of training, the accuracy of the model reached 93.15%, the recall rate reached 99.96%, and the average accuracy rate (mAP) reached 99.46%. According to the detection effect of the model on the test set, the impact of training times, different light sources and different shooting distances on the model’s detection effect was explored. The statistical results showed that the model detection accuracy rate was the highest under the green light source, and the image detection accuracy rate was the highest under the shooting height of 5cm. The research results were also compared with YOLO v5s, RetinaNet and YOLO v4 network models in 50 iterations. The results showed that the mAP of YOLO v5-MDC model was 99.40%, which was 0.06 percentage points lower than that of the original YOLO v5s model, but the model occupied the smallest storage space, with a result of only 13.4MB, which was 0.6 MB less than the YOLO v5s model. The average detection time for single image was 0.03s, and the maximum detection time was 0.08s. In summary, the designed model can effectively realize the detection of heavily adhesive wheat grains. At the same time, the model had high detection efficiency and small storage space, which can provide necessary technical support for the development of embedded equipment for wheat grain detection.

    参考文献
    相似文献
    引证文献
引用本文

宋怀波,王云飞,段援朝,宋磊,韩梦璇.基于YOLO v5-MDC的重度粘连小麦籽粒检测方法[J].农业机械学报,2022,53(4):245-253. SONG Huaibo, WANG Yunfei, DUAN Yuanchao, SONG Lei, HAN Mengxuan. Detection Method of Severe Adhesive Wheat Grain Based on YOLO v5-MDC Model[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(4):245-253.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-04-07
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2021-07-06
  • 出版日期: