基于改进YOLOv4的无人机目标检测方法
2021-09-23邹金霖赵晓林
田 港, 张 鹏, 邹金霖, 赵晓林
(空军工程大学装备管理与无人机工程学院,西安, 710051)
目标检测已经在车辆检测、人脸检测、自动驾驶、安全系统等领域广泛应用并发挥着重要的作用。相比于传统检测方法,基于深度学习的检测模型具有检测精度高、速度快等优点。基于深度学习的检测模型有one-stage和two-stage两种:one-stage检测模型主要有YOLO[1]、SSD[2]、DSSD[3]等模型,其核心是基于回归的思想,不需要使用区域候选网络,可以直接经主干网络预测出军事目标类别、位置;以two-stage检测模型为代表的有R-CNN[4]、SPP-Net[5]、Fast R-CNN[6]、Faster R-CNN[7]、Mask R-CNN[8]等模型,其基本思想是通过卷积神经网络对图像进行特征提取,然后经过区域候选网络生成候选框,再经过“抠图”和进一步的特征表示,最后就可以通过分类和回归得到军事目标的类别和位置。two-stage系列因其繁琐的检测过程导致检测速度远不能满足实时性的要求。
近年来,目标检测在战场感知中的地位愈加突出。基于无人机平台的目标检测具有视野大、监测范围广等优点,但由于载重、能耗等方面的限制,无人机平台能够搭载的存储和算力是有限的。因此,一般检测模型在无人机上直接部署时会面临模型占用内存大、实时性差的问题。因此,在保证检测精度的前提下有效解决上述问题,能够有效提升无人机侦察的检测速度,有助于战场指挥员或机载系统及时全面分析战场形势,做出更合理的决策。
YOLO作为one-stage检测模型中的经典模型,较好地兼顾了检测精度和速度,因此应用较为广泛。本文以YOLOv4[9]检测模型作为主模型,在保证检测精度的情况下,首先,将三尺度检测模型改进为双尺度检测模型。……
