|国家科技期刊平台
首页|期刊导航|舰船电子工程|基于改进YOLOv5的军事目标识别方法

基于改进YOLOv5的军事目标识别方法OACSTPCD

A Military Target Detection Method Based on Improved YOLOv5

中文摘要英文摘要

针对战场环境下因背景干扰和军事目标尺度较小等原因导致误检、漏检的问题,提出一种基于改进YOLOv5的军事目标识别方法CB-YOLOv5.利用坐标注意力机制重构特征提取主干网络,增强网络对复杂背景下军事目标的特征提取能力;在特征融合网络中引入BiFPN,减少浅层特征信息的丢失,提高对弱小目标的检测能力.在自建数据集下实验表明,改进后算法mAP达到93.8%,比原模型提升了3.5%,可以有效识别战场环境下多尺度军事目标.

To address the problem of false detection and missed detection due to background interference and small scale of military targets in battlefield environment,a military target recognition method CB-YOLOv5 based on improved YOLOv5 is pro-posed.The feature extraction backbone network is reconstructed by using coordinate attention mechanism to enhance the feature ex-traction capability of the network for military targets in complex background.BiFPN is introduced in the feature fusion network to re-duce the loss of shallow feature information and improve the weak targets can be detected.Experiments under the self-built dataset show that the improved algorithm mAP reaches 93.8%,which is 3.5%better than the original model,and can effectively identify multi-scale military targets in the battlefield environment.

万晓刚;王伟

西安工程大学计算机科学学院 西安 710600

计算机与自动化

目标识别YOLOv5注意力机制特征融合

target detectionYOLOv5attention mechanismfeature fusion

《舰船电子工程》 2024 (004)

28-33 / 6

2021年中国高校产学研创新基金项目(编号:2021ALA02002)资助.

10.3969/j.issn.1672-9730.2024.04.007

评论