基于模型压缩的ED-YOLO电力巡检无人机避障目标检测算法

作者：彭继慎; 孙礼鑫^*; 王凯; 宋立业

来源：仪器仪表学报, 2021, 42(10): 161-170.

DOI：10.19650/j.cnki.cjsi.J2108173

• 电力巡检无人机
• 目标检测
• 注意力机制
• 深度可分离卷积
• 模型压缩

针对现有卷积神经网络模型体积大、运算量高,导致电力巡检无人机检测速率与精度无法兼顾的问题,提出一种基于模型压缩的ED-YOLO网络实现无人机避障的目标检测算法。该目标检测算法以YOLOv4为基础,首先在主干网络中加入通道注意力机制,在不增加计算量前提下提高检测精度;其次在特征金字塔部分运用深度可分离卷积替换传统卷积,减少卷积计算量;最后利用模型压缩策略裁剪网络中冗余通道,减小模型体积并提高模型检测速度。在自主构建的9 600张电力巡检无人机飞行障碍的数据集进行测试,ED-YOLO与YOLOv4相比,其障碍物目标检测的平均精度均值只降低了1.4%,而模型体积减少了94.9%,浮点运算量减少了82.1%,预测速度提升了2.3倍。实验结果表明,对比多种其他现存方法,本文提出的基于模型压缩的ED-YOLO目标检测算法有着精度高、体积小和检测速度快的优势,满足电力巡检无机避障检测要求。

• 单位
  辽宁工程技术大学