基于改进YOLOv3-tiny的田间行人与农机障碍物检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

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


Detection of Pedestrian and Agricultural Vehicles in Field Based on Improved YOLOv3-tiny
Author:
Affiliation:

Fund Project:

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

    为实现农机自主作业中的避障需求,本文针对室外田间自然场景中因植被遮挡、背景干扰而导致障碍物难以检测的问题,基于嵌入式平台应用设备,提出了农机田间作业时行人和农机障碍物检测的改进模型,更好地平衡了模型的检测速度与检测精度。该改进模型以You only look once version 3 tiny(YOLOv3-tiny)为基础框架,融合其浅层特征与第2 YOLO预测层特征作为第3预测层,通过更小的预选框增加小目标表征能力;在网络关键位置的特征图中混合使用注意力机制中的挤压激励注意模块(Squeeze and excitation attention module,SEAM) 与卷积块注意模块(Convolutional block attention module,CBAM),通过强化检测目标关注以提高抗背景干扰能力。建立了室外环境下含农机与行人的共9405幅图像的原始数据集。其中训练集7054幅,测试集2351幅。测试表明本文模型的内存约为YOLOv3与单次多重检测器(Single shot multibox detector,SSD)模型内存的1/3和2/3;与YOLOv3-tiny相比,本文模型平均准确率(Mean average precision,mAP)提高11个百分点,小目标召回率(Recall)提高14百分点。在Jetson TX2嵌入式平台上本文模型的平均检测帧耗时122ms,满足实时检测要求。

    Abstract:

    The real-time detection of pedestrian and agricultural vehicles is very important for the navigation and path planning of autonomous agricultural vehicles. In the field, obstacles are difficult to be detected due to crops occlusion and background interference. A real-time pedestrian and agricultural vehicles detection model in natural field scene was proposed, which effectively improved the feasibility of pedestrian and agricultural vehicles visual detection to embedded platform in the independent operation of agricultural machinery. This detection model was improved based on You only look once version 3 tiny (YOLOv3-tiny). A third prediction layer was got by merging the features of YOLOv3-tiny’s shallow layer and the features of second YOLO prediction layer, thus more smaller anchors resulted in the detection ability improvement of small targets. Both the squeeze and excitation attention module (SEAM) and the convolutional block attention module (CBAM) were applied in the key feature maps of the network, thus the model’s anti-background disturbance capability was increased. A data set included 9405 images of pedestrian and agricultural vehicles with different shooting angles and natural field scenes was set, and 7054 images were used for training while the remained 2351 images were used for testing. Tests showed that the memory size of the improved model was reduced to 1/3 and 2/3 of that of the YOLOv3 and single shot multibox detector (SSD) models, the improved model’s mean average precision (mAP) was increased by 11 percentage points, and the small target recall (R) rate was increased by 14 percentage points while compared with that of YOLOv3-tiny. On the Jetson TX2 embedded hardware platform, the single frame detection time of the improved model was 122ms, which can meet the requirements of real-time detection.

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

李文涛,张岩,莫锦秋,李彦明,刘成良.基于改进YOLOv3-tiny的田间行人与农机障碍物检测[J].农业机械学报,2020,51(s1):1-8,33. LI Wentao, ZHANG Yan, MO Jinqiu, LI Yanming, LIU Chengliang. Detection of Pedestrian and Agricultural Vehicles in Field Based on Improved YOLOv3-tiny[J]. Transactions of the Chinese Society for Agricultural Machinery,2020,51(s1):1-8,33.

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