Ship Detection Based on an Enhanced YOLOv5 Algorithm
Journal Title: Mechatronics and Intelligent Transportation Systems - Year 2024, Vol 3, Issue 4
Abstract
Advanced ship detection technologies play a critical role in improving maritime safety by enabling the rapid identification of vessels and other maritime targets, thereby mitigating the risk of collisions and optimizing traffic efficiency. Traditional detection methods often demonstrate high sensitivity to minor variations in target appearance but face significant limitations in generalization, making them ill-suited to the complex and dynamic nature of maritime environments. To address these challenges, an enhanced ship detection method, referred to as YOLOv5-SE, has been proposed, which builds upon the YOLOv5 framework. This approach incorporates attention mechanisms within the backbone network to improve the model's focus on key features of small targets, dynamically adjusting the importance of each channel to boost representational capacity and detection accuracy. In addition, a refined version of the Complete Intersection over Union (CIoU) loss function has been introduced to optimize the loss associated with target bounding box prediction, thereby improving localization accuracy and ensuring more precise alignment between predicted and ground-truth boxes. Furthermore, the conventional coupled detection head in YOLOv5 is replaced by a Decoupled Head, facilitating better adaptability to various target shapes and accelerating model convergence. Experimental results demonstrate that these modifications significantly enhance ship detection performance, with mean Average Precision (mAP) at IoU 0.5 reaching 94.9% and 95.1%, representing improvements of 3.1% and 1.2% over the baseline YOLOv5 model, respectively. These advancements underscore the efficacy of the proposed methodology in improving detection accuracy and robustness in challenging maritime settings.
Authors and Affiliations
Xin Liu, Qingfa Zhang, Yubo Tu, Mingzhi Shao, Tengwen Zhang, Yuhan Sun, Haiwen Yuan
Keywords
Related Articles
- EP ID EP755203
- DOI https://doi.org/10.56578/mits030403
- Views 33
- Downloads 0
Assessing Automatic Dependent Surveillance-Broadcast Signal Quality for Airplane Departure Using Random Forest Algorithm
This study aims to assess the safety level of the Automatic Dependent Surveillance-Broadcast (ADS-B) signal quality during airplane departures at Sultan Mahmud Badaruddin II Airport. The Aero-track application was utiliz...
Stability Simulation and Analysis of Maglev Vehicle at Different Speed Based on UM
High-speed maglev train is an excellent mode of transportation to build a transportation network because of its high safety, low emission, low energy consumption, low noise and strong climbing ability. The safety, stabil...
Strategic Selection of Crowd Logistics Platforms: A Multi-Criteria Decision-Making Approach
Crowd logistics (CL) represents an innovative model within the logistics sector, leveraging the participation of individuals to enhance service provision, optimize resource utilization, and reduce operational costs. Amon...
Optimising Dual Pantograph-Catenary System Performance in Curved Sections for Enhanced High-Speed Railway Operations
The spatial configuration of the pantograph-catenary system (PCS) is significantly altered by the superelevation present in curved railway tracks, leading to deviations in the system’s dynamic behaviour and imposing cons...
Regional Transformation via Rail: A Historical and Analytical Examination of Iran's Railway Network and its Socio-Economic Impacts
The role of transport infrastructure, especially railways, in shaping a nation's socio-economic and cultural dynamics is of paramount importance. The present research delves into the profound influence of the railway net...