|国家科技期刊平台
首页|期刊导航|光学精密工程|融合特征重组和注意力的小样本战斗部破片群目标检测

融合特征重组和注意力的小样本战斗部破片群目标检测OA北大核心CSTPCD

中文摘要

战斗部破片群运动参数对弹药毁伤威力评估具有重要的意义。针对破片尺寸较小、背景信息复杂以及破片数据样本少导致的破片检测精度较低的问题,本文提出一种YOLOv5-FD的战斗部破片群目标检测方法。首先,在网络输出端增加微小目标检测层,将原始的三尺度改为四尺度,并在特征融合网络中引入内容感知特征重组(Content Aware ReAssembly of Features,CARAFE)上采样模块替换原有的最近邻插值上采样,减少小目标特征信息损失,提高弱小破片的提取能力。其次,在特征提取网络引入坐标注意力模块(Coordinate Attention,CA),加强对破片特征的提取,弱化背景信息,抑制复杂背景的干扰。最后,在模型训练过程中引入模型不可知元学习方法(Model Agnostic Meta Learning,MAML),达到仅用小样本破片数据集实现较高的检测性能。实验结果表明,YOLOv5-FD破片检测算法在自制破片数据集中,精确率达到了90.5%,召回率达到了85.4%,平均精度mAP_0.5达到了88.2%,与原始YOLOv5s算法相比分别提高了7.1%,7.9%和7.5%,有效提高了破片目标检测准确性。

和萌;武江鹏;梁超;胡鹏宇;任媛;贺翾;刘强辉;

西安现代控制技术研究所,陕西西安710065

计算机与自动化

高速破片目标检测YOLOv5特征重组注意力机制元学习

《光学精密工程》 2024 (012)

P.1929-1940 / 12

基础加强计划重点项目(No.2019-JCJQ-ZD-360-12);国防基础科研计划项目(No.JCKY2021208B011)。

10.37188/OPE.20243212.1929

评论