系统工程与电子技术 ›› 2023, Vol. 45 ›› Issue (5): 1270-1276.doi: 10.12305/j.issn.1001-506X.2023.05.02

• 电子技术 • 上一篇    

基于YOLOv5的光学遥感图像舰船目标检测算法

成倩, 李佳, 杜娟   

  1. 空军工程大学基础部, 陕西 西安 710038
  • 收稿日期:2022-03-15 出版日期:2023-04-21 发布日期:2023-04-28
  • 通讯作者: 成倩
  • 作者简介:成倩(1990—), 女, 讲师, 硕士, 主要研究方向为计算机视觉、模式识别
    李佳(1977—), 女, 副教授, 博士, 主要研究方向为光电成像、图像处理
    杜娟(1990—), 女, 讲师, 博士, 主要研究方向为图像超分辨重建

Ship target detection algorithm of optical remote sensing image based on YOLOv5

Qian CHENG, Jia LI, Juan DU   

  1. Fundamentals Department, Air Force Engineering University, Xi'an 710038, China
  • Received:2022-03-15 Online:2023-04-21 Published:2023-04-28
  • Contact: Qian CHENG

摘要:

针对YOLO(you only look once)v5算法在应用于光学遥感图像舰船目标检测任务时所面临的小目标误检率、漏检率较高的情况, 提出一种基于YOLOv5改进的光学遥感图像舰船目标检测方法。首先对路径聚合网络结构进行改进, 设计语义信息增强模块提取更富语义信息的浅层特征, 增强对小目标特征的表达能力; 然后使用Swish函数作为激活函数, 提高网络对数据非线性特征的表征能力, 加快模型的收敛速度; 最后针对舰船目标的尺寸特点优化检测端结构, 移除大目标检测头以减少推理计算量。测试集上的实验表明,该方法相较改进前将检测精度提高了5.2%且推理时间有所减少, 在保证检测实时性的同时增强了模型的小目标辨别能力。

关键词: 光学遥感图像, 舰船目标检测, YOLOv5, 语义信息增强, 激活函数

Abstract:

When applied to the ship target detection task of optical remote sensing images, you only look once (YOLO)v5 algorithm suffered false positive and false negative a lot on small targets. An improved method based on YOLOv5 is proposed to deal with this situation. Firstly, a module named semantic information enhancement module aiming at extracting shallow features with more semantic information is designed and added into the path aggregation network, so as to enhance the expression ability of small target features. Then, the Swish function is used as the activation function to improve the network's ability to characterize the nonlinear characteristics of the data and accelerate the convergence speed of the model. Finally, according to the size characteristics of the ship, the large target detection head is removed to reduce the amount of inference calculation and optimize the effect of the detection end. The experiment on the test set shows that compared with the method before the improvement, the proposed method improves the detection accuracy by 5.2% and the inference time is reduced. At the same time of ensuring the real-time detection, the small target discrimination ability of the model is increased.

Key words: optical remote sensing image, ship target detection, YOLO(you only look once)v5, semantic information enhancement, activation function

中图分类号: