基于改进YOLOv5的潜艇目标检测算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

陕西省科技厅基金项目(No.2023YBGY369)资助。


Submarine target detection algorithm based on improved YOLOv5
Author:
Affiliation:

Fund Project:

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

    为了解决传统潜艇目标检测缺乏对复杂背景和噪声的鲁棒性、对光照变化和视角变化敏感、难以处理大规模数据集等问题,提出了一种基于改进YOLOv5潜艇目标检测器。通过C3_Transformer结构,有效提升了特征的全局上下文建模能力和长距离依赖性捕捉能力;通过simOTA解决anchorbased算法中正负样本不平衡问题,增强模型对小目标和困难样本的学习能力;利用decoupledhead的思想解决分类和位置预测任务的互斥性问题,提高检测精度和鲁棒性。实验结果表明:相较于原始YOLOv5,改进后的模型Precision、Recall、mAP05、mAP 05∶095分别提高了28、109、38、147,这表明改进后的模型在潜艇目标检测的准确性、召回率以及在不同置信度阈值下的平均准确率等方面取得了明显的进步,同时在实际检测任务中改进后的模型有效解决了“漏检”、“误检”的问题。

    Abstract:

    In this paper,an improved YOLOv5 based submarine target detector is proposed to solve the problems that traditional submarine target detection lacks robustness to complex backgrounds and noises,is sensitive to changes in illumination and viewing angle,and is difficult to deal with large scale datasets.With the C3_Transformer structure,the global context modeling ability of the features and the long range dependency capturing ability are effectively improved.And the simOTA algorithm is employed to address the issue of imbalanced positive and negative samples in anchor based algorithms,thereby enhancing the model′s learning capabilities for small targets and challenging samples.Additionally,the decoupledhead approach is utilized to overcome the mutual exclusivity problem between classification and position prediction tasks,resulting in improved detection accuracy and robustness.The experimental results show that compared to the original YOLOv5,the improved model shows significant advancements in terms of Precision,Recall,mAP@0.5,and mAP@0.5∶0.95,with improvements of 2.8%,10.9%,3.8%,and 14.7% respectively,which indicates that the improved model achieves notable progress in terms of accuracy,recall rate,and average precision at different confidence thresholds in submarine target detection.Furthermore,the improved model effectively addresses the issues of "missed detection" and "false detection" in the actual detection task.

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

梅礼坤,陈智利.基于改进YOLOv5的潜艇目标检测算法[J].激光与红外,2024,54(6):971~979
MEI Li-kun, CHEN Zhi-li. Submarine target detection algorithm based on improved YOLOv5[J]. LASER & INFRARED,2024,54(6):971~979

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