一种基于改进YOLOv4的SAR舰船检测算法
DOI:
CSTR:
作者:
作者单位:

1.内蒙古科技大学信息工程学院 内蒙古包头 014010;2.大连海事大学信息科学与技术学院 辽宁大连116026

作者简介:

通讯作者:

中图分类号:

TP751

基金项目:


A SAR Ship Detection Algorithm Based on Improved YOLOv4
Author:
Affiliation:

1.School of Information Engineering, Inner Mongolia University of Science and Technology, Baotou 014010, China 2.Information Science and Technology College, Dalian Maritime University, Dalian 116026, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    在大量数据支持的背景下,如何高效利用大量SAR图像,提升舰船目标的检测精度是当前舰船目标检测的难题。本文聚焦如何提升YOLOv4算法对SAR舰船目标的检测精度,提出了一种融合多尺度和注意力增强的YOLOv4增强算法。在原YOLOv4的PANet中加入注意力模块(CBAM),同时使用加强的K-means聚类算法对数据集中的舰船目标真实框进行聚类,并对锚框结果进行线性比例变换,让算法锚框更适合于训练集。实验证明本文提出的算法在SAR舰船检测中的平均准确率()达到了94.05%,比原始YOLOv4精度提高了0.7%。实验结果充分证明本文提出的算法能够提升SAR舰船图像检测精度,为海上活动判断精确化提供技术支持。

    Abstract:

    Under the background of large amount of data support, how to use a large number of SAR images efficiently and improve the accuracy of ship target detection is the current problem of ship target detection. This paper focuses on how to improve the accuracy of YOLOv4 algorithm for SAR ship target detection, and presents a YOLOv4 enhancement algorithm that combines multiscale and attention enhancement. The Attention Module (CBAM) is added to the PANet of the original YOLOv4, and the enhanced K-means clustering algorithm is used to cluster the ship target real frame in the dataset, and the result of the anchor frame is transformed linearly to make the algorithm anchor frame more suitable for the training set. Experiments show that the average accuracy () of the proposed algorithm in SAR ship detection is 94.05%, which is 0.7% higher than that of the original YOLOv4. The experimental results fully demonstrate that the proposed algorithm can improve the accuracy of SAR ship image detection and provide technical support for the accuracy of sea activities judgment.

    参考文献
    相似文献
    引证文献
引用本文

陈洋,张明,杨立东,喻大华,张宝华,李建军.一种基于改进YOLOv4的SAR舰船检测算法[J].电子测量技术,2022,45(11):120-125

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-25
  • 出版日期:
文章二维码