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 条
  • [41] EC-YOLO: Improved YOLOv7 Model for PCB Electronic Component Detection
    Luo, Shiyi
    Wan, Fang
    Lei, Guangbo
    Xu, Li
    Ye, Zhiwei
    Liu, Wei
    Zhou, Wen
    Xu, Chengzhi
    SENSORS, 2024, 24 (13)
  • [42] Object Detection Based on Improved YOLOv7 for UAV Aerial Image
    Cui, Liqun
    Cao, Huawei
    Computer Engineering and Applications, 60 (20): : 189 - 197
  • [43] YOLOv7-sea: Object Detection of Maritime UAV Images based on Improved YOLOv7
    Zhao, Hangyue
    Zhang, Hongpu
    Zhao, Yanyun
    2023 IEEE/CVF WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WORKSHOPS (WACVW), 2023, : 233 - 238
  • [44] Tea leaf disease detection and identification based on YOLOv7 (YOLO-T)
    Md. Janibul Alam Soeb
    Md. Fahad Jubayer
    Tahmina Akanjee Tarin
    Muhammad Rashed Al Mamun
    Fahim Mahafuz Ruhad
    Aney Parven
    Nabisab Mujawar Mubarak
    Soni Lanka Karri
    Islam Md. Meftaul
    Scientific Reports, 13
  • [45] Tea leaf disease detection and identification based on YOLOv7 (YOLO-T)
    Soeb, Md. Janibul Alam
    Jubayer, Md. Fahad
    Tarin, Tahmina Akanjee
    Al Mamun, Muhammad Rashed
    Ruhad, Fahim Mahafuz
    Parven, Aney
    Mubarak, Nabisab Mujawar
    Karri, Soni Lanka
    Meftaul, Islam Md.
    SCIENTIFIC REPORTS, 2023, 13 (01)
  • [46] BPN-YOLO: A Novel Method for Wood Defect Detection Based on YOLOv7
    Wang, Rijun
    Chen, Yesheng
    Liang, Fulong
    Wang, Bo
    Mou, Xiangwei
    Zhang, Guanghao
    FORESTS, 2024, 15 (07):
  • [47] BD-YOLO: High-Precision Lightweight Concrete Bubble Detector Based on YOLOv7
    Chen, Kun
    Wang, Anzhi
    PATTERN RECOGNITION AND COMPUTER VISION, PT III, PRCV 2024, 2025, 15033 : 379 - 393
  • [48] Improved YOLOv7 for UAV Image Object Detection
    Zou, Zhentao
    Li, Zeping
    Computer Engineering and Applications, 60 (08): : 173 - 181
  • [49] EP-YOLO: A Pipeline Leak Detection Algorithm Based on Improved YOLOv7
    Zhang, Jiali
    Lan, Xiaowen
    Wang, Shaofeng
    Liu, Wenjing
    IEEE ACCESS, 2024, 12 : 198132 - 198141
  • [50] Fs-yolo: fire-smoke detection based on improved YOLOv7
    Wang, Dongmei
    Qian, Ying
    Lu, Jingyi
    Wang, Peng
    Hu, Zhongrui
    Chai, Yongkang
    MULTIMEDIA SYSTEMS, 2024, 30 (04)