Improving the underwater object detection algorithm for YOLOv5

Jingtao Yang; Lihui Sun; Gang Wang

doi:10.1117/12.3032864

5 July 2024 Improving the underwater object detection algorithm for YOLOv5

Jingtao Yang, Lihui Sun, Gang Wang

Proceedings Volume 13184, Third International Conference on Electronic Information Engineering and Data Processing (EIEDP 2024); 1318477 (2024) https://doi.org/10.1117/12.3032864
Event: 3rd International Conference on Electronic Information Engineering and Data Processing (EIEDP 2024), 2024, Kuala Lumpur, Malaysia

Abstract

This paper addresses the challenges of target occlusion and the difficulty in detecting small underwater organisms due to the behavioral characteristics of underwater fauna. An improved YOLOv5 algorithm for underwater target detection is proposed to enhance performance in underwater environments. Firstly, YOLOv5 incorporates the Coordinate Attention module to further enhance feature extraction. Secondly, a BiFPN (Bidirectional Feature Pyramid Network) is employed to better integrate features of different scales. Finally, a DyHead (Detection Head based on attention mechanism) is added for target detection. Experimental results on the URPC2020 dataset show that the improved YOLOv5 network achieves a 1.7% increase in mAP@0.5 and a 1.1% increase in mAP@0.5-0.95. The parameter count of the improved YOLOv5 network only increases by 5.5%, with a computational overhead of 5.6%. Therefore, the proposed approach demonstrates superior performance in underwater target detection tasks.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Jingtao Yang, Lihui Sun, and Gang Wang "Improving the underwater object detection algorithm for YOLOv5", Proc. SPIE 13184, Third International Conference on Electronic Information Engineering and Data Processing (EIEDP 2024), 1318477 (5 July 2024); https://doi.org/10.1117/12.3032864

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
7 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Detection and tracking algorithms

Target detection

Submerged target modeling

Feature fusion

Head

Environmental sensing

Show All Keywords

Keywords/Phrases

Search In:

Publication Years