基于重参数化注意力的无人机航拍目标检测方法

作     者：彭晏飞 陈炎康 赵涛 袁晓龙 陈坤 

作者机构：辽宁工程技术大学电子与信息工程学院 

出 版 物：《电光与控制》 (Electronics Optics & Control)

年 卷 期：2024年

学科分类：08[工学] 080203[工学-机械设计及理论] 082503[工学-航空宇航制造工程] 0802[工学-机械工程] 0825[工学-航空宇航科学与技术] 

基　　金：国家自然科学基金（61772249） 辽宁省高等学校基本科研项目（LJKZ0358，LJKQZ2021152） 辽宁工程技术大学双一流学科创新团队资助项目（LNTU20TD-27） 

主　　题：目标检测 无人机 重参数化 注意力机制 解耦检测头 

摘      要：针对无人机航拍图像目标尺度变化大、目标与背景相互干扰所导致的误检、漏检问题，提出一种基于重参数化注意力的目标检测方法，并将其应用于无人机航拍目标检测。首先，通过提出重参数化坐标注意力模块来增强相关特征，提升网络对上下文信息的捕捉能力；其次，设计多尺度感受野增强模块来重构骨干网络，从而增强特征图的接受域，提升网络的特征提取能力；接着，提出四尺度特征融合检测网络，提升网络对小目标的检测能力；最后，引入解耦检测头来解决分类与回归任务之间的冲突。在VisDrone数据集上进行实验，mAP0.5相比原算法提高了7.6%，召回率提升了5.5%，与其他方法相比也具有明显优势。实验结果证明，改进方法能够较好地解决无人机航拍图像中目标尺度变化大、目标与背景相互干扰所带来的错检、漏检问题，具有良好的检测效果。