文献类型：期刊文章

作　　者：奉志强[1] 谢志军[1] 包正伟[2] 陈科伟[3]

FENG Zhiqiang;XIE Zhijun;BAO Zhengwei;CHEN Kewei(School of Information Science and Engineering,Ningbo University,Ningbo 315211,China;Ningbo JIWANG Information Technology Ltd,Ningbo 315000,China;School of Mechanical Engineering and Mechanics,Ningbo University,Ningbo 315211,China)

机构地区：[1]宁波大学信息科学与工程学院,宁波315211 [2]宁波极望信息科技有限公司,宁波315000 [3]宁波大学机械工程与力学学院,宁波315211

出　　处：《航空学报》

基　　金：国家自然科学基金(U20A20121);浙江省自然基金(LY21F020006);宁波市自然科学基金(2019A610088);宁波市“科技创新2025”重大专项(2019B10125,2019B10028,20201ZDYF020077)。

年　　份：2023

卷　　号：44

期　　号：7

起止页码：246-260

语　　种：中文

收录情况：AJ、BDHX、BDHX2020、CAS、CSCD、CSCD2023_2024、EAPJ、EI、IC、JST、RCCSE、SCOPUS、ZGKJHX、核心刊

摘　　要：无人机航拍图像与自然场景图像相比背景更复杂,存在大量密集小目标,对检测网络提出了更高的要求。在保证目标检测实时性的前提下,针对无人机视角下密集小目标检测精度低的问题,提出一种基于YOLOv5的无人机实时密集小目标检测算法。首先,将空间注意力(SAM)与通道注意力(CAM)相结合,改进CAM中特征压缩后的全连接层,降低计算量。另外,改变CAM与SAM的连接结构,提高空间维度特征捕获能力。综上,提出一种空间-通道注意力模块(SCAM),提高模型对特征图中小目标聚集区域的关注程度;其次,提出一种基于SCAM的注意力特征融合模块(SC-AFF),根据不同尺度特征图自适应分配注意力权重,增强小目标的特征融合效率;最后,在主干网络中引入Trans⁃former模块,并利用SC-AFF模块改进原有的残差连接处的特征融合方式,更好地捕获全局信息和丰富的上下文信息提高复杂背景下密集小目标的特征提取能力。在VisDrone2021数据集上进行实验,YOLOv5s基准下,改进后模型的mAP50提高了6.4%,mAP75提高了5.8%,对高分辨率图像的FPS可达到46。在输入分辨率1504×1504下训练的模型mAP50可达54.5%,比YOLOv4提高了11.5%,精度提高的同时检测速度FPS依旧保持在46,更适用于密集小目标场景下的无人机实时目标检测。

关 键 词：无人机 小目标检测 注意力机制  自注意力机制  特征融合  

分 类 号：V279] TP391.4] TP183[计算机类]