基于孪生网络的红外无人机小目标辅助检测方法

Infrared UAV Small Target Auxiliary Detection Method Based on Siamese Network

  • 摘要: 各种非管控小型无人机对公共安全造成一定的威胁。文中基于红外探测技术设计一种轻量化无人机小目标辅助检测算法(auxiliary detection methods,ADM)。提出一种全新的辅助训练方式,该方法通过定义两个具有相同模型结构和初始化参数的骨干网络以达到辅助训练的效果;本文在Ghost PAN的基础上进行结构改进,构建一种更适合无人机目标检测的多尺度特征融合结构。消融实验结果表明本文算法所涉及的各个模块对无人机目标检测精度(mAP)皆有提升;多算法对比实验结果表明本文算法能够适应多种不同的无人机飞行场景,与Nanodet Plus-m相比检测时间基本不变,在Sea、Sky、Mountain和City数据集场景中,mAP分别提升11.4%、4.2%、16%和4.2%。

     

    Abstract: Small uncontrolled drones pose threats to public safety. In this study, we designed a lightweight unmanned aerial vehicle target detection algorithm based on infrared detection technology. This method achieves the effect of auxiliary training by defining two backbone networks with the same model structure and initialization parameters. This paper improves the structure on the basis of Ghost PAN to build a multi-scale feature fusion structure that is more suitable for UAV target detection. The ablation experiment results show that each module involved in this algorithm improves the UAV target detection accuracy (), and the results of the multi-algorithm comparison experiments show that the algorithm proposed in this study can adapt to a variety of UAV flight scenarios. Compared to Nanodet Plus-m, the detection time was unchanged, and the mAP increased by 11.4%, 4.2%, 16%, and 4.2% on the Sea, Sky, Mountain, and City datasets, respectively.

     

/

返回文章
返回