Modern Defense Technology ›› 2020, Vol. 48 ›› Issue (6): 81-88.DOI: 10.3969/j.issn.1009-086x.2020.06.012

• DETECTION AND TRACKING TECHNOLOGY • Previous Articles     Next Articles

Air-to-Ground Target Intelligent Detection Algorithm Based on Dual Attention Mechanism

WANG Wen-qing1, PANG Ying1, LIU Yang1, YANG Dong-fang2, ZHANG Meng2   

  1. 1. Xi'an University of Posts & Telecommunications,Automation College,Shaanxi Xi'an 710121,China;
    2. Rocket Force of Engineering University,Missile Engineering College,Shaanxi Xi'an 710025,China
  • Received:2020-01-16 Revised:2020-06-09 Online:2020-12-20 Published:2021-02-01

双重注意机制的空对地目标智能检测算法

王文庆1, 庞颖1, 刘洋1, 杨东方2, 张萌2   

  1. 1.西安邮电大学 自动化学院,陕西 西安 710121;
    2.火箭军工程大学 导弹工程学院,陕西 西安 710025
  • 作者简介:王文庆(1964-),男,北京房山人。教授,博士,主要从事复杂系统结构分析与鲁棒控制,智能信息处理,信息系统分析、设计与开发等方面的教学科研工作。通信地址:710121 陕西省西安市长安区西安邮电大学 E-mail: wwq@xupt.edu.cn
  • 基金资助:
    国家自然科学基金(61673017,61403398);陕西省自然科学基金(2017JM6077);陕西省重点研发计划项目(2018ZDXM-GY-039)

Abstract: Air-to-ground target detection plays an important role in performing typical military and civilian tasks on air-based unmanned platforms such as drones and missiles.Aiming at the problems of fewer effective features and high false detection rate of the air-to-ground target intelligent detection method,an air-to-ground target intelligent detection algorithm based on the dual attention mechanism is proposed.First,a channel attention mechanism is established,focusing on the important channels in the feature layer that contain the effective features of the target.Second,the spatial attention mechanism is used to enable the network to focus on local key target areas.The dual attention mechanism combines the advantages of the channel attention mechanism and the spatial attention mechanism to improve the efficiency of target detection.A loss function that couples the dual attention mechanism and target detection is also designed,which realizes the synchronous optimization of the loss function.Finally,a comparative experiment is performed using the air-to-ground target detection data set.The experimental results show that the proposed algorithm has higher target detection accuracy and training speed than other algorithms in the air-to-ground scenario.

Key words: air-to-ground target detection, deep learning, complex feature interference, dual attention mechanism, effective feature, key area

摘要: 空对地目标检测在无人机、导弹等空基无人平台执行典型军事和民用任务中均发挥着重要作用。针对空对地目标检测存在目标有效特征少和误检测率高等问题,提出了一种基于双重注意机制的空对地目标智能检测算法。首先,建立通道注意机制,重点关注特征层中包含目标有效特征的重要通道;其次,利用空间注意机制,关注局部重点目标区域;而双重注意机制结合了通道注意机制和空间注意机制的优势,提升目标检测效率。还设计了双重注意机制和目标检测两者相耦合的损失函数,实现了损失函数的同步优化;最后,利用空对地目标检测数据集进行了对比实验。实验结果表明,在空对地场景下,所提出的算法与其他算法相比,具有更高目标检测精度和训练速度。

关键词: 空对地目标检测, 深度学习, 复杂特征干扰, 双重注意机制, 有效特征, 重点区域

CLC Number: