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] Lightweight Underwater Target Detection Algorithm Based on YOLOv7
    Xin, Shiao
    Ge, Haibo
    Yuan, Hao
    Yang, Yudi
    Yao, Yang
    Ma, Sai
    2024 6TH INTERNATIONAL CONFERENCE ON NATURAL LANGUAGE PROCESSING, ICNLP 2024, 2024, : 387 - 391
  • [2] Study on Lightweight Model of Maize Seedling Object Detection Based on YOLOv7
    Zhao, Kai
    Zhao, Lulu
    Zhao, Yanan
    Deng, Hanbing
    APPLIED SCIENCES-BASEL, 2023, 13 (13):
  • [3] 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):
  • [4] 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
  • [5] YOLO-ME: an enhanced lightweight YOLOv7 Tiny model for efficient object detection in aerial imagery
    Junos, Mohamad Haniff
    Zulkifli, Safiah
    Bakar, Elmi Abu
    Hawary, Ahmad Faizul
    Khairuddin, Anis Salwa Mohd
    SIGNAL IMAGE AND VIDEO PROCESSING, 2025, 19 (04)
  • [6] Improved Underwater Object Detection Algorithm of YOLOv7
    Liang, Xiuman
    Li, Ran
    Yu, Haifeng
    Liu, Zhendong
    Computer Engineering and Applications, 2024, 60 (06) : 89 - 99
  • [7] 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
  • [8] Improved YOLOv7 Underwater Object Detection Based on Attention Mechanism
    Fu, Junshang
    Tian, Ying
    ENGINEERING LETTERS, 2024, 32 (07) : 1377 - 1384
  • [9] Underwater optical image object detection based on YOLOv7 algorithm
    Wang, Shaojie
    Wu, Weichao
    Wang, Xinyuan
    Han, Yongchen
    Ma, Yuwei
    OCEANS 2023 - LIMERICK, 2023,
  • [10] Improved Lightweight Underwater Target Detection Algorithm of YOLOv7
    Xin, Shi'ao
    Ge, Haibo
    Yuan, Hao
    Yang, Yudi
    Yao, Yang
    Computer Engineering and Applications, 2024, 60 (03)