-
题名结合仿真迁移学习和自适应融合的无人机小目标检测
被引量：4
- 1
-
-
作者
陈蕊
郑华飞
蒋鸿宇
郭有为
-
机构
中国工程物理研究院研究生院
中国工程物理研究院电子工程研究所
-
出处
《小型微型计算机系统》
CSCD
北大核心
2023年第8期1743-1749,共7页
-
基金
中国工程物理研究院军民融合科研发展基金项目(Y21K-TR01)资助。
-
文摘
无人机小目标的精确检测在公共安全和无人机防御系统中起着至关重要的作用.被广泛应用于通用目标检测任务的深度学习技术,在无人机小目标检测任务上的效果往往受限于稀缺的相关数据资源以及较小的目标尺度.针对以上问题,本文提出了一种基于仿真迁移学习和自适应融合机制相结合的无人机小目标检测方法.该方法首先利用基于UnrealEngine的Air-Sim仿真平台生成丰富且高保真的无人机小目标仿真图像数据,以减轻对稀缺真实图像数据的依赖.其次,为解决仿真图像与真实图像的数据分布差异问题,应用模型参数知识迁移技术,首先在仿真数据集上YOLOv5目标检测模型进行预训练,随后利用真实数据集对模型进行微调训练.最后,为进一步适应小目标检测场景,提出了一种基于YOLOv5的改进神经网络模型AF-YOLO,该网络引入了自适应融合机制.实验结果表明,基于仿真的迁移学习方法效果优于基准方法,使无人机目标检测的性能提升2.7%;引入自适应融合机制的方法,使性能提升6.2%;最终,基于仿真迁移学习和自适应融合机制相结合的方法与基准方法相比,性能提升7.1%.
-
关键词
无人机小目标检测
无人机仿真数据
迁移学习
YOLOv5
自适应特征融合
-
Keywords
UAV small object detection
UAV simulation data generation
transfer learning
YOLO v5
adaptive fusion
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-
