基于特征聚合与多元协同特征交互的航拍图像小目标检测  

Small object detection in aerial images based on feature aggregation and multiple cooperative features interaction

作　　者：陈朋磊 王江涛[1,2] 张志伟 何程 Chen Penglei;Wang Jiangtao;Zhang Zhiwei;He Cheng(School of Physics and Electronic Information,Huaibei Normal University,Huaibei 235000,China;Ahui Province Key Laboratory of Intelligent Computing and Applications,Huaibei 235000,China)

机构地区：[1]淮北师范大学物理与电子信息学院,淮北235000 [2]智能计算及应用安徽省重点实验室,淮北235000

出　　处：《电子测量与仪器学报》2023年第10期183-192,共10页Journal of Electronic Measurement and Instrumentation

基　　金：国家自然科学基金(61976101);安徽省高校自然科学研究重点项目(2023AH050319);安徽省高校优秀科研创新团队项目(2023AH010044)资助。

摘　　要：针对无人机航拍图像目标尺寸太小、包含的特征信息较少,导致现有的检测算法对小目标检测效果不理想的问题,提出一种基于特征聚合与多元协同特征交互的无人机航拍图像小目标检测算法。首先,针对主干网对特征提取不足的问题,采用Swin Transformer作为RetinaNet主干网络,以增强算法对全局信息的提取能力。其次,为提高网络对远处目标即小目标的检测能力,设计出一种高效的小目标特征聚合网络(SFANet),实现对浅层特征图小目标细节信息的充分整合。最后,为进一步提高网络对多尺度目标的检测性能,使低层特征信息流向高层,提出全新的多元协同特征交互模板(MCFIM)。在公开无人机航拍数据集VisDrone2019-DET上的实验结果表明,所提算法相较于原RetinaNet基线网络检测精度提高7.6%,对于小目标具有更好的检测效果。Aiming at the problem that the target size of the UAV aerial image is too small and contains less feature information,which leads to the unsatisfactory detection effect of the existing detection algorithm on small objects,a UAV aerial photography based on feature aggregation and multi-collaborative feature interaction is proposed.First of all,in view of the insufficient feature extraction of the backbone network,Swin Transformer is selected as the RetinaNet backbone network to enhance the global information extraction ability of the algorithm.Secondly,in order to improve the detection ability of remote targets,a small target feature aggregation network is proposed,which can fully integrate the details of small targets in shallow feature maps.Finally,in order to further improve the detection performance of multi-scale targets,a new multiple collaborative feature interaction module is proposed to make the low-level feature information flow to the high-level.Experimental results on VisDrone2019-DET,a public UAV aerial photo data set,show that compared with the original RetinaNet baseline network detection precision increased by 7.6%,the proposed algorithm has better detection effect for small targets.

关 键 词：小目标检测 航拍图像 小目标特征聚合网络 多元协同特征交互模块 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]