GCP-YOLO: a lightweight underwater object detection model based on YOLOv7

被引:2
|
作者
Gao, Yu [1 ]
Li, Zhanying [1 ]
Zhang, Kangye [1 ]
Kong, Lingyan [1 ]
机构
[1] Dalian Polytech Univ, Sch Informat Sci & Engn, Dalian 116034, Peoples R China
关键词
YOLOv7; Underwater object detection; Prune; CA;
D O I
10.1007/s11554-024-01586-7
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
With the rapid development of technology, underwater biological detection tasks are generally conducted using mobile devices. This paper proposes the GCP-YOLO model based on YOLOv7 to address the challenges of deploying large detection models on mobile devices in the field of underwater object detection, particularly the issues of difficulty in underwater object detection and resource constraints. First, the GhostNetV2 module is used to make the ELAN module lightweight in the Neck part, reducing the model's parameter count and computational complexity. Second, to address potential issues such as feature loss and low accuracy when collecting feature information in the lightweight network, we incorporate the CA Attention module after the improved ELAN module to prevent feature loss caused by the lightweighting process. Finally, we perform pruning on the overall improved model with a pruning rate of 50%\documentclass[12pt]{minimal} \usepackage{amsmath} \usepackage{wasysym} \usepackage{amsfonts} \usepackage{amssymb} \usepackage{amsbsy} \usepackage{mathrsfs} \usepackage{upgreek} \setlength{\oddsidemargin}{-69pt} \begin{document}$$\%$$\end{document}, further reducing the model's parameter count and computational complexity. Compared to the YOLOv7 model, the GCP-YOLO underwater object detection model reduces the parameter count and computational complexity by a factor of 4, while improving accuracy by 2.8%\documentclass[12pt]{minimal} \usepackage{amsmath} \usepackage{wasysym} \usepackage{amsfonts} \usepackage{amssymb} \usepackage{amsbsy} \usepackage{mathrsfs} \usepackage{upgreek} \setlength{\oddsidemargin}{-69pt} \begin{document}$$\%$$\end{document}.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] Bi2F-YOLO: a novel framework for underwater object detection based on YOLOv7
    Xiaopeng Liu
    Keke Zhao
    Cong Liu
    Long Chen
    Intelligent Marine Technology and Systems, 3 (1):
  • [2] Research on Underwater Object Detection Algorithm Based on YOLOv7
    Shi, Biying
    Zhang, Lianbo
    Tang, Jialin
    Yan Jinghui
    2024 CROSS STRAIT RADIO SCIENCE AND WIRELESS TECHNOLOGY CONFERENCE, CSRSWTC 2024, 2024, : 501 - 506
  • [3] Improved YOLOv7 model for underwater sonar image object detection
    Qin, Ken Sinkou
    Liu, Di
    Wang, Fei
    Zhou, Jingchun
    Yang, Jiaxuan
    Zhang, Weishi
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2024, 100
  • [4] An Underwater Target Detection Algorithm Based on Attention Mechanism and Improved YOLOv7
    Ren, Liqiu
    Li, Zhanying
    He, Xueyu
    Kong, Lingyan
    Zhang, Yinghao
    CMC-COMPUTERS MATERIALS & CONTINUA, 2024, 78 (02): : 2829 - 2845
  • [5] YOLOv7-CHS: An Emerging Model for Underwater Object Detection
    Zhao, Liang
    Yun, Qing
    Yuan, Fucai
    Ren, Xu
    Jin, Junwei
    Zhu, Xianchao
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (10)
  • [6] A Lightweight Model of Underwater Object Detection Based on YOLOv8n for an Edge Computing Platform
    Fan, Yibing
    Zhang, Lanyong
    Li, Peng
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2024, 12 (05)
  • [7] Hcl-yolo: a lightweight and efficient underwater object detection algorithm
    Xiuman Liang
    Teng Zhang
    Haifeng Yu
    Zhendong Liu
    Journal of Real-Time Image Processing, 2025, 22 (2)
  • [8] CEH-YOLO: A composite enhanced YOLO-based model for underwater object detection
    Feng, Jiangfan
    Jin, Tao
    ECOLOGICAL INFORMATICS, 2024, 82
  • [9] BiGA-YOLO: A Lightweight Object Detection Network Based on YOLOv5 for Autonomous Driving
    Liu, Jun
    Cai, Qiqin
    Zou, Fumin
    Zhu, Yintian
    Liao, Lyuchao
    Guo, Feng
    ELECTRONICS, 2023, 12 (12)
  • [10] Rcf-yolo: an underwater object detection algorithm based on improved YOLOv10n
    Xiuman Liang
    Teng Zhang
    Haifeng Yu
    Zhendong Liu
    Journal of Real-Time Image Processing, 2025, 22 (2)