基于改进YOLO v4的落叶松毛虫侵害树木实时检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

黑龙江省自然科学基金联合引导项目(LH2020C049)


Real-time Detection Method of Dendrolimus superans-infested Larix gmelinii Trees Based on Improved YOLO v4
Author:
Affiliation:

Fund Project:

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

    针对two-stage网络模型训练成本高,无人机搭载的边缘计算设备检测速度低等问题,提出一种基于改进YOLO v4模型的受灾树木实时检测方法,以提高对落叶松毛虫虫害树木的识别精度与检测速度。以黑龙江省大兴安岭地区呼玛县白银纳乡受落叶松毛虫侵害的落叶松无人机图像为数据,利用LabelImg软件标注75~100m的无人机图像,构建落叶松毛虫虫害树木图像数据集。将CSPNet应用于YOLO v4模型的Neck架构,重新设计Backbone的特征提取网络——CSPDarknet53模型结构,并在CSPNet进行优化计算前的卷积中加入SENet以增加感受野信息,使其改变网络的深度、宽度、分辨率及网络结构,实现模型缩放,提高检测精度。同时,在PANet中使用CSPConvs卷积代替原有卷积Conv×5,最后经过YOLO Head检测输出预测结果。将YOLO v4-CSP网络模型部署至GPU进行训练,训练过程的内存降低至改进前的82.7%。再搭载至工作站进行测试,结果表明:改进的YOLO v4-CSP网络模型在测试阶段对虫害树木检测的正确率为97.50%,相比于YOLO v4的平均正确率提高3.4个百分点,模型精度接近目前主流two-stage框架Faster R-CNN的98.75%;将改进的YOLO v4-CSP网络模型搭载至Jetson nano边缘计算设备,检测速度达到4.17f/s,高于YOLO v4模型的1.72f/s。基于YOLO v4-CSP的检测模型可实现对受灾树木检测速度与精度的平衡,降低模型的应用成本,搭载至无人机可实现对森林虫害的实时监测。

    Abstract:

    Aiming at the problems of high training cost of two-stage network model and low detection speed of edge computing equipment attached on UAV, a real-time detection method based on the improved YOLO v4 model was proposed in order to improve the recognition accuracy and detection speed for Dendrolimus superans-infested Larix gmelinii trees. Taking the UAV images of Larix gmelinii infested by Dendrolimus superansobtained from Baiyinna Township, Huma County in the Daxing'anling District of Heilongjiang Province as data, the UAV images at 75~100m were marked with LabelImg software, and a data set of tree images infested by Dendrolimus superanswas constructed. CSPNet was applied to the Neck architecture of the YOLO v4 model, the Backbones feature extraction network—CSPDarknet53 model structure was redesigned, and SENet was added to the convolution before CSPNet optimization calculations to increase the receptive field information, making it change the depth, width, resolution and structure of the network to achieve model scaling and improve detection accuracy. Meanwhile, CSPConvs convolution was used in PANet to replace the original convolution Conv×5, and finally the prediction result was output through YOLO Head detection. After deploying the YOLO v4-CSP network model to the GPU for training, the memory of the training process was reduced to 82.7% of that before improvement. The improved model was installed on the workstation for testing. Results showed that the accuracy of tree detection was 97.50%, which was 3.4 percentage points higher than the average detection accuracy of YOLO v4, and close to 98.75% of the current mainstream two-stage framework Faster R-CNN. When attached to Jetson nano edge computing equipment, the detection speed was 4.17f/s, higher than the 1.72f/s of YOLO v4 model. Therefore, the proposed detection model based on YOLO v4-CSP can achieve balance between detection speed and detection accuracy for the Dendrolimus superans-infested Larix gmelinii trees, reduce application cost of the model, and realize real-time monitoring of forest pests when attached to UAV.

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

林文树,张金生,何乃磊.基于改进YOLO v4的落叶松毛虫侵害树木实时检测方法[J].农业机械学报,2023,54(4):304-312,393. LIN Wenshu, ZHANG Jinsheng, HE Nailei. Real-time Detection Method of Dendrolimus superans-infested Larix gmelinii Trees Based on Improved YOLO v4[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(4):304-312,393.

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