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

作     者：彭晏飞 陈炎康 赵涛 袁晓龙 陈坤 PENG Yanfei;CHEN Yankang;ZHAO Tao;YUAN Xiaolong;CHEN Kun

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

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

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

年 卷 期：2024年第31卷第9期

页      码：81-86,110页

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

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

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 

D　O　I：10.3969/j.issn.1671-637X.2024.09.014

馆 藏 号：203138917...