Open Access iconOpen Access

ARTICLE

crossmark

An Underwater Target Detection Algorithm Based on Attention Mechanism and Improved YOLOv7

Liqiu Ren, Zhanying Li*, Xueyu He, Lingyan Kong, Yinghao Zhang

College of Information Science and Engineering, Dalian Polytechnic University, Dalian, 116034, China

* Corresponding Author: Zhanying Li. Email: email

(This article belongs to the Special Issue: Machine Vision Detection and Intelligent Recognition)

Computers, Materials & Continua 2024, 78(2), 2829-2845. https://doi.org/10.32604/cmc.2024.047028

Abstract

For underwater robots in the process of performing target detection tasks, the color distortion and the uneven quality of underwater images lead to great difficulties in the feature extraction process of the model, which is prone to issues like error detection, omission detection, and poor accuracy. Therefore, this paper proposed the CER-YOLOv7(CBAM-EIOU-RepVGG-YOLOv7) underwater target detection algorithm. To improve the algorithm’s capability to retain valid features from both spatial and channel perspectives during the feature extraction phase, we have added a Convolutional Block Attention Module (CBAM) to the backbone network. The Reparameterization Visual Geometry Group (RepVGG) module is inserted into the backbone to improve the training and inference capabilities. The Efficient Intersection over Union (EIoU) loss is also used as the localization loss function, which reduces the error detection rate and missed detection rate of the algorithm. The experimental results of the CER-YOLOv7 algorithm on the UPRC(Underwater Robot Prototype Competition) dataset show that the mAP(mean Average Precision) score of the algorithm is 86.1%, which is a 2.2% improvement compared to the YOLOv7. The feasibility and validity of the CER-YOLOv7 are proved through ablation and comparison experiments, and it is more suitable for underwater target detection.

Keywords


Cite This Article

APA Style
Ren, L., Li, Z., He, X., Kong, L., Zhang, Y. (2024). An underwater target detection algorithm based on attention mechanism and improved yolov7. Computers, Materials & Continua, 78(2), 2829-2845. https://doi.org/10.32604/cmc.2024.047028
Vancouver Style
Ren L, Li Z, He X, Kong L, Zhang Y. An underwater target detection algorithm based on attention mechanism and improved yolov7. Comput Mater Contin. 2024;78(2):2829-2845 https://doi.org/10.32604/cmc.2024.047028
IEEE Style
L. Ren, Z. Li, X. He, L. Kong, and Y. Zhang "An Underwater Target Detection Algorithm Based on Attention Mechanism and Improved YOLOv7," Comput. Mater. Contin., vol. 78, no. 2, pp. 2829-2845. 2024. https://doi.org/10.32604/cmc.2024.047028



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 365

    View

  • 200

    Download

  • 0

    Like

Share Link