基于协同注意力与特征融合的无人机小目标检测

作者:韩晶; 王伟宇; 吕学强*; 陈玉忠; 赵海兴; 才藏太
来源:北京信息科技大学学报(自然科学版), 2023, 38(03): 1-8.
DOI:10.16508/j.cnki.11-5866/n.2023.03.001

摘要

针对现有无人机检测领域存在小目标检测困难以及特征提取效果不佳的问题,提出了一种基于协同注意力模块与双向特征融合结构的无人机小目标检测网络。通过设计协同注意力单元,与骨干网络形成协同注意力模块,显著提升模型特征表达能力;同时借鉴路径聚合网络的思想,设计双向特征融合结构,引入自底向上的特征融合路径,实现浅层特征与深层特征的深度融合,提高小目标检测精度。实验结果表明,所提算法在无人机小目标数据集上的平均精度优于其他网络,显著提升了无人机场景下小目标的检测性能。

全文