欢迎访问《兵工学报》官方网站,今天是 分享到:

兵工学报 ›› 2022, Vol. 43 ›› Issue (2): 345-354.doi: 10.3969/j.issn.1000-1093.2022.02.012

• 论文 • 上一篇    下一篇

改进的YOLOv3算法及其在军事目标检测中的应用

于博文, 吕明   

  1. (南京理工大学 自动化学院, 江苏 南京 210094)
  • 上线日期:2022-03-28
  • 通讯作者: 吕明(1980—),女,副研究员,博士 E-mail:18761654456@163.com
  • 作者简介:于博文(1988—), 男, 博士研究生。E-mail: yubowen_0403@njust.edu.cn
  • 基金资助:
    江苏省自然科学基金项目(BK20180467)

Improved YOLOv3 Algorithm and Its Application in Military Target Detection

YU Bowen, L Ming   

  1. (School of Automation, Nanjing University of Science and Technology, Nanjing 210094, Jiangsu, China)
  • Online:2022-03-28

摘要: 复杂环境下军事目标检测技术是提高战场态势生成、分析能力的基础和关键。针对军事目标检测任务在复杂环境下传统检测算法的检测性能较低问题,提出一种基于改进YOLOv3的军事目标检测算法,通过深度学习实现复杂环境下军事目标的自动检测。构建军事目标图像数据集,为各类目标检测算法提供测试环境;在网络结构上通过引入可形变卷积改进的ResNet50-D残差网络作为特征提取网络,提高网络对形变目标的检测精度和速度;在特征融合阶段引入双注意力机制和特征重构模块,增强目标特征的表征能力,抑制干扰,提升检测精度;利用DIOU损失函数和Focal损失函数重新设计目标检测器的损失函数,进一步提高其对军事目标的检测精度;在军事目标图像数据集中进行测试实验。实验结果表明,改进的YOLOv3算法相比于原YOLOv3算法,平均精度均值提高了2.98%,检测速度提高了8.6帧/s,具有较好的检测性能,可为战场态势生成、分析提供有效的辅助技术支持。

关键词: 目标检测, 可形变卷积, YOLOv3算法, 特征融合, 注意力机制

Abstract: Military target detection in a complex environment is the basis and key to improving battlefield situation generation and analysis capability. For the military target detection tasks, the detection performance of traditional detection algorithms in complex environment is low. A military target detection algorithm based on improved YOLOv3 algorithm is proposed to automatically detect the military targets in complex environment through deep learning. A military target image dataset is constructed to provide a testing environment for various target detection algorithms. The detection accuracy and speed of deformable target are improved by introducing the deformable convolutional improved ResNet50-D residual network as feature extraction network. In the stage of feature fusion, a dual-attention mechanism and feature reconstruction module are introduced to enhance the characterization ability of target features, suppress the interference, and improve the detection accuracy. The loss function of target detector is redesigned by using DIOU Loss functions and Focal Loss to funther improve the detection accuracy of military targets. The experimental results show that the improved YOLOv3 algorithm improves the average detection accuracy by 2.98% and the detection speed by 8.6 frames/s compared with the original YOLOv3 algorithm. The improved YOLOv3 algorithm has better detection performance and can provide effective auxiliary technical support for battlefield situation generation and analysis.

Key words: targetdetection, deformableconvolution, YOLOv3algorithm, featurefusion, attentionmechanism

中图分类号: