结合数据增强和改进 YOLOv4 的水下目标检测算法
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391. 4

基金项目:

国家自然科学基金(61801169,61873086)、中央高校基本科研业务费(B220202020)项目资助


Underwater object detection algorithm combining data enhancement and improved YOLOv4
Author:
Affiliation:

Fund Project:

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

    针对水下低质量成像、水下目标形态大小各异、以及水下目标重叠或遮挡导致水下目标检测精度低的问题,提出一种结 合数据增强和改进 YOLOv4(you look only once)的水下目标检测算法,在 YOLOv4 的主干特征提取网络 CSPDarknet53 中添加卷 积块注意力机制(convolutional block attention module,CBAM),以提高网络模型特征提取能力;在路径聚合网络( path aggregation network,PANet)中添加同层跳接和跨层跳接结构,以增强网络模型多尺度特征融合能力;通过数据增强方法 PredMix(predictionmix)模拟水下生物重叠、遮挡等显示不完全的情形,以增强网络模型鲁棒性。 实验结果表明,结合数据增强和改进 YOLOv4 的 水下目标检测算法在 URPC2018(underwater robot picking control 2018)数据集上的检测精度提升到了 78. 39%,比 YOLOv4 高出 7. 03%,充分证明所提算法的有效性。

    Abstract:

    Aiming at the problem of low underwater object detection accuracy caused by low-quality underwater imaging, different shapes or sizes of underwater objects, and overlapping or occlusion of underwater objects, an underwater object detection algorithm combining data enhancement and improved YOLOv4 is proposed. By adding CBAM ( convolutional block attention module) to the backbone of YOLOv4—CSPDarknet53, the feature extraction ability of network model is improved. In order to enhance the multi-scale feature fusion ability, the same-layer skip connections and cross-layer skip connections are added to PANet (path aggregation network). To enhance the robustness of the network model, the data enhancement method PredMix (prediction mix) is used to simulate the incomplete display of underwater organisms such as overlap or occlusion. The experimental results show that the detection accuracy of the underwater object detection algorithm combining data enhancement and improved YOLOv4 on URPC2018 dataset is improved to 78. 39%, 7. 03% higher than YOLOv4, which fully proves the effectiveness of the proposed algorithm.

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

史朋飞,韩 松,倪建军,杨 鑫.结合数据增强和改进 YOLOv4 的水下目标检测算法[J].电子测量与仪器学报,2022,36(3):113-121

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