English

中国农机化学报

中国农机化学报 ›› 2023, Vol. 44 ›› Issue (7): 200-206.DOI: 10.13733/j.jcam.issn.2095-5553.2023.07.027

• 农业智能化研究 • 上一篇    下一篇

基于YOLOv5模型的飞蓬属入侵植物目标检测

李宗南,蒋怡,王思,李源洪,黄平,魏鹏   

  1. 四川省农业科学院遥感与数字农业研究所,成都市,610066
  • 出版日期:2023-07-15 发布日期:2023-07-31
  • 基金资助:
    四川省科技计划项目(2021YFG0028);国家重点研发计划子课题(2021YFD1600803)

Object detection of invasive Erigeron L. plants base on YOLOv5

Li Zongnan, Jiang Yi, Wang Si, Li Yuanhong, Huang Ping, Wei Peng   

  • Online:2023-07-15 Published:2023-07-31

摘要: 为应用深度学习模型实现机器快速准确识别农田恶性杂草,以田间常见的2种飞蓬属入侵植物为对象,采集样本图像并标注杂草目标,基于网络结构深度、宽度可调的一阶段目标检测模型YOLOv5搭建训练平台和嵌入式测试平台,训练14组具有不同网络层和卷积核的模型权重,验证模型精度及检测帧率。结果表明:不同网络结构深度、宽度设置的YOLOv5模型识别飞蓬属入侵植物的平均精度为91.8%~95.1%,有8组权重的平均精度优于YOLOv3的,合理增加网络层和卷积核能提高模型精度;YOLOv5在训练平台的帧率为28~109fps之间,在测试平台的帧率为12~58fps之间,有12组权重的帧率比YOLOv3的有显著提高,帧率受平台算力限制并随网络层和卷积核增加而下降,在算力较低的嵌入式系统中实现实时检测需平衡模型网络结构的设置。该研究结果可为搭建农田杂草智能感知系统提供参考。

关键词: 深度学习, 目标检测, 卷积神经网络, 入侵植物, 杂草, 智慧农业

Abstract: In order to achieve fast and accurate identification of malignant weeds in agricultural fields using deep learning models, this study took two common invasive plants of Erigeron L. in the fields, collected sample images and labeled weed targets, and built a training platform and an embedded test platform based on YOLOv5 with adjustable depth and width of network structure. Fourteen groups of model weights with different network layers and convolution kernels were trained, and the accuracy and detection frame rate were evaluated. The results showed that the average precision of YOLOv5 with different network structure depth and width settings for identifying invasive plants ranged from 91.8% to 95.1%. Eight groups of weights achieved higher average precision than YOLOv3, indicating that the increase of network layers and convolution kernels can improve the model accuracy. The results also showed that frame rates of YOLOv5 were between 28 to 109 fps on the training platform and between 12 to 58 fps on the test platform. Twelve sets of weights exhibited significantly improved frame rates compared to YOLOv3. The frame rate is limited by the computing capability of the platform and decreases with the increase of network layers and convolution kernels. To realize realtime detection in the embedded system with low compute capability, a balanced network structure setting is needed. The results of this study can provide a reference for building an intelligent weed sensing system in agricultural fields.

Key words: deep learning, object detection, CNN, invasive plants, weeds, smart agriculture

中图分类号: