MarineYOLO: Innovative deep learning method for small target detection in underwater environments
Document Type
Article
Publication Date
10-1-2024
Abstract
In the realm of underwater object detection, conventional methodologies often encounter challenges in accurately identifying and detecting small targets. These difficulties stem primarily from the intricate nature of underwater environments, suboptimal lighting conditions, and the diminutive scale of the targets themselves. To address this persistent challenge, the MarineYOLO network is introduced. This approach involves refining the conventional C2f module into the EC2f module, alongside the integration of the Efficient Multi-scale Attention (EMA) module into YOLOv8. Additionally, the Convolutional Block Attention Module (CBAM) is introduced to further refine the Feature Pyramid Network (FPN), facilitating enhanced feature extraction pertinent to small targets. Furthermore, the conventional CIoU is replaced with Wise-IoU to augment the precision and stability of target localization. Experimental findings demonstrate that MarineYOLO achieves an average precision (AP) of 78.5% on the RUOD dataset and 88.1% on the URPC dataset, marking improvements of 12.2% and 16.8%, respectively, compared to YOLOv8n. As an emerging paradigm in underwater object detection, MarineYOLO harbors significant potential in both practical applications and scholarly endeavors, furnishing an efficacious remedy to the challenges associated with detecting small targets in underwater settings.
Keywords
Underwater object detection, MarineYOLO, CBAM, Wise-IoU, EMA
Divisions
fsktm
Funders
Key Research Project Universities of Henan Province (24A520061),Science and Technology Plan Project
Publication Title
Alexandria Engineering Journal
Volume
104
Publisher
Elsevier
Publisher Location
RADARWEG 29, 1043 NX AMSTERDAM, NETHERLANDS