航空学报 > 2022, Vol. 43 Issue (5): 325270-325270   doi: 10.7527/S1000-6893.2021.25270

基于多尺度深度学习的自适应航拍目标检测

刘芳, 韩笑   

  1. 北京工业大学 信息学部, 北京 100124
  • 收稿日期:2021-01-14 修回日期:2021-02-08 发布日期:2021-03-01
  • 通讯作者: 韩笑 E-mail:hxdeemail@163.com
  • 基金资助:
    国家自然科学基金(61171119)

Adaptive aerial object detection based on multi-scale deep learning

LIU Fang, HAN Xiao   

  1. Faculty of Information Technology, Beijing University of Technology, Beijing 100124, China
  • Received:2021-01-14 Revised:2021-02-08 Published:2021-03-01
  • Supported by:
    National Natural Science Foundation of China (61171119)

摘要: 无人机已经被广泛应用到各个领域,目标检测成为无人机视觉领域关键技术之一。针对无人机图像中场景复杂、尺度多变、小目标丰富等问题,提出了一种基于多尺度深度学习的自适应航拍目标检测算法。首先,构建自适应特征提取网络MSDarkNet-53,引入多尺度卷积方式,采用不同类型卷积核对不同尺寸目标进行运算,有效扩大感受野。其次,结合注意力机制的优点设计卷积模块,自适应优化特征权重,增强有效特征,抑制无效特征,得到表征能力更强的特征。最后,构建基于多尺度特征融合的预测网络,根据小目标的特点,选取多层级特征映射融合成高分辨率特征图,在单一尺度上进行目标分类和边界框回归。实验表明:本文算法提升了无人机图像的目标检测精度,具有良好的鲁棒性。

关键词: 无人机, 目标检测, 多尺度卷积, 注意力机制, 特征融合

Abstract: Unmanned Aerial Vehicles (UAV) have been widely used in various fields, and object detection has become one of the key technologies in the field of UAV vision. Considering the problems of complex scenes, variable scales and too many small targets in UAV images, an adaptive aerial object detection algorithm is proposed based on multi-scale convolution. Firstly, the multi-scale feature extraction network is constructed, and the multi-scale convolution method is introduced. Different types of convolution are used to check different sizes of targets to expand the receptive field effectively. Secondly, the convolution module is designed according to the advantages of attention mechanism, and the feature weight is adaptively optimized to obtain more representational features. Finally, a prediction network is constructed based on multi-scale feature fusion. According to the characteristics of small targets, multi-level feature maps are selected to fuse into high-resolution feature maps, and object classification and boundary box regression are carried out on a single scale. Experimental results show that the proposed algorithm improves the object detection accuracy of UAV aerial images, and has good robustness.

Key words: Unmanned Aerial Vehicles (UAV), object detection, multi-scale convolution, attention mechanism, feature fusion

中图分类号: