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 条
  • [31] ASSD-YOLO: a small object detection method based on improved YOLOv7 for airport surface surveillance
    Zhou, Wentao
    Cai, Chengtao
    Zheng, Liying
    Li, Chenming
    Zeng, Daohui
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (18) : 55527 - 55548
  • [32] ASSD-YOLO: a small object detection method based on improved YOLOv7 for airport surface surveillance
    Wentao Zhou
    Chengtao Cai
    Liying Zheng
    Chenming Li
    Daohui Zeng
    Multimedia Tools and Applications, 2024, 83 : 55527 - 55548
  • [33] PBA-YOLOv7: An Object Detection Method Based on an Improved YOLOv7 Network
    Sun, Yang
    Li, Yi
    Li, Song
    Duan, Zehao
    Ning, Haonan
    Zhang, Yuhang
    APPLIED SCIENCES-BASEL, 2023, 13 (18):
  • [34] Rapid detection of Yunnan Xiaomila based on lightweight YOLOv7 algorithm
    Wang, Fenghua
    Jiang, Jin
    Chen, Yu
    Sun, Zhexing
    Tang, Yuan
    Lai, Qinghui
    Zhu, Hailong
    FRONTIERS IN PLANT SCIENCE, 2023, 14
  • [35] Maize Kernel Quality Detection Based on Improved Lightweight YOLOv7
    Yang, Lili
    Liu, Chengman
    Wang, Changlong
    Wang, Dongwei
    AGRICULTURE-BASEL, 2024, 14 (04):
  • [36] DSML-YOLO: A Fast and Lightweight Underwater Object Detection Model Based on Sonar Images
    Li, Guangyu
    Liao, Xiaofei
    Jin, Yan
    Chao, Bohang
    2024 IEEE 4TH INTERNATIONAL CONFERENCE ON SOFTWARE ENGINEERING AND ARTIFICIAL INTELLIGENCE, SEAI 2024, 2024, : 112 - 116
  • [37] Underwater-YCC: Underwater Target Detection Optimization Algorithm Based on YOLOv7
    Chen, Xiao
    Yuan, Mujiahui
    Yang, Qi
    Yao, Haiyang
    Wang, Haiyan
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (05)
  • [38] Lightweight YOLOv7 for bushing surface defects detection
    Cheng, Wenjun
    Zeng, Pengfei
    Hao, Yongping
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (02)
  • [39] Enhanced YOLOv7 for Improved Underwater Target Detection
    Lu, Daohua
    Yi, Junxin
    Wang, Jia
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2024, 12 (07)
  • [40] Yolo-tla: An Efficient and Lightweight Small Object Detection Model based on YOLOv5
    Ji, Chun-Lin
    Yu, Tao
    Gao, Peng
    Wang, Fei
    Yuan, Ru-Yue
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)