基于改进YOLO v5s的马铃薯种薯芽眼检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

山东省薯类产业技术体系农业机械岗位专家项目(SDAIT-16-10)和中国博士后科学基金项目(2020M681690)


Detection Method of Potato Seed Bud Eye Based on Improved YOLO v5s
Author:
Affiliation:

Fund Project:

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

    芽眼检测是马铃薯种薯智能切块首先要解决的问题,为实现种薯芽眼精准高效检测,提出了一种基于改进YOLO v5s的马铃薯种薯芽眼检测方法。首先通过加入CBAM注意力机制,加强对马铃薯种薯芽眼图像的特征学习和特征提取,同时弱化与芽眼相似的马铃薯种薯表面背景对检测结果的影响。其次引入加权双向特征金字塔BiFPN增加经骨干网络提取的种薯芽眼原始信息,为不同尺度特征图赋予不同权重,使得多尺度特征融合更加合理。最后替换为改进的高效解耦头Decoupled Head区分回归和分类,加快模型收敛速度,进一步提升马铃薯种薯芽眼检测性能。试验结果表明,改进YOLO v5s模型准确率、召回率和平均精度均值分别为93.3%、93.4%和95.2%;相比原始YOLO v5s模型,平均精度均值提高3.2个百分点,准确率、召回率分别提高0.9、1.7个百分点;不同模型对比分析表明,改进YOLO v5s模型与Faster R-CNN、YOLO v3、YOLO v6、YOLOX和YOLO v7等模型相比有着较大优势,平均精度均值分别提高8.4、3.1、9.0、12.9、4.4个百分点。在种薯自动切块芽眼检测试验中,改进YOLO v5s模型平均召回率为91.5%,相比原始YOLO v5s模型提高17.5个百分点。本文方法可为研制马铃薯种薯智能切块芽眼识别装置提供技术支持。

    Abstract:

    The first problem to be solved in potato cutting fast is the detection of potato seed bud eyes, an improved YOLO v5s-based potato seed bud eye detection method was proposed to improve seed potato eye detection performance. Firstly, by adding the CBAM attention mechanism, the feature learning and feature extraction of the potato bud eye images were strengthened. The influence of the potato surface background similar to the bud eyes on the detection results was weakened. Secondly, the weighted bidirectional feature pyramid BiFPN was introduced to increase the original information of bud eyes extracted by the backbone network and assign weights to feature maps of different scales, making multi-scale feature fusion more reasonable. Finally, it was replaced with an improved and efficient Decoupled Head to distinguish between regression and classification, speed up the convergence speed of the model, and further improve the performance of potato bud eye detection. The test results showed that the precision, recall rate, and average precision of the improved algorithm were 93.3%, 93.4% and 95.2%, respectively, which was 3.2 percentage points higher than that of the original algorithm in the mean average precision, and the precision and recall rate were improved by 0.9 and 1.7 percentage points. The comparative analysis of different algorithms showed that this algorithm had absolute advantages compared with Faster R-CNN, YOLO v3, YOLO v6,YOLOX and YOLO v7 algorithms. The mAP was increased by 8.4 percentage points, 3.1 percentage points, 9.0 percentage points,12.9 percentage points and 4.4 percentage points. In the actual detection application, the average recall rate of the improved algorithm was 91.5%, which was 17.5 percentage points higher than that of the original algorithm, and the missed detection rate was reduced. The method can provide technical support for the next step in the development of a sprout-eye identification device for the intelligent cutting of potato seed potatoes.

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

张万枝,曾祥,刘树峰,穆桂脂,张弘毅,郭壮壮.基于改进YOLO v5s的马铃薯种薯芽眼检测方法[J].农业机械学报,2023,54(9):260-269. ZHANG Wanzhi, ZENG Xiang, LIU Shufeng, MU Guizhi, ZHANG Hongyi, GUO Zhuangzhuang. Detection Method of Potato Seed Bud Eye Based on Improved YOLO v5s[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(9):260-269.

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