Research on Underwater Small Target Detection Algorithm Based on Improved YOLOv7

被引:19
作者
Yi, Weiguo [1 ]
Wang, Bo [1 ]
机构
[1] Dalian Jiaotong Univ, Sch Comp & Commun Engn, Dalian 116028, Peoples R China
关键词
YOLOv7; underwater small target detection; attention mechanism; FPN; EIoU;
D O I
10.1109/ACCESS.2023.3290903
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Target detection research has always been difficult when it comes to small target detection in underwater situations. To address the issues of a high miss detection rate and poor underwater scene recognition in underwater small target detection tasks, an improved underwater small target detection technique utilizing YOLOv7 is proposed. To achieve the accuracy rate while considering the high detection speed, the YOLOv7 network is used as the basic network. The network concentrates more crucial feature information of small targets to increase detection accuracy while reducing model complexity by merging the SENet attention mechanism, enhancing the FPN network topology, and incorporating the EIoU loss function. Through simulation tests, the mAP, P, and R metrics are confirmed on the test set and contrasted with other conventional target detection techniques. The outcomes demonstrate that the enhanced algorithm outperforms competing networks and successfully raises detection accuracy on the test set.
引用
收藏
页码:66818 / 66827
页数:10
相关论文
共 42 条
  • [1] Benjumea A, 2021, Arxiv, DOI arXiv:2112.11798
  • [2] Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
  • [3] Underwater Target Recognition Based on Improved YOLOv4 Neural Network
    Chen, Lingyu
    Zheng, Meicheng
    Duan, Shunqiang
    Luo, Weilin
    Yao, Ligang
    [J]. ELECTRONICS, 2021, 10 (14)
  • [4] Underwater object detection using Invert Multi-Class Adaboost with deep learning
    Chen, Long
    Liu, Zhihua
    Tong, Lei
    Jiang, Zheheng
    Wang, Shengke
    Dong, Junyu
    Zhou, Huiyu
    [J]. 2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
  • [5] Cheng G, 2023, Arxiv, DOI arXiv:2207.14096
  • [6] Davis J., 2006, P 23 INT C MACH LEAR
  • [7] Research Challenges, Recent Advances, and Popular Datasets in Deep Learning-Based Underwater Marine Object Detection: A Review
    Er, Meng Joo
    Chen, Jie
    Zhang, Yani
    Gao, Wenxiao
    [J]. SENSORS, 2023, 23 (04)
  • [8] Ge Z, 2021, Arxiv, DOI arXiv:2107.08430
  • [9] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [10] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587