Improved YOLOv8n for Lightweight Ship Detection
Zhiguang Gao,Xiaoyan Yu,Xianwei Rong,Wenqi Wang
DOI: https://doi.org/10.3390/jmse12101774
IF: 2.744
2024-10-06
Journal of Marine Science and Engineering
Abstract:Automatic ship detection is a crucial task within the domain of maritime transportation management. With the progressive success of convolutional neural networks (CNNs), a number of advanced CNN models have been presented in order to detect ships. Although these detection models have achieved marked performance, several undesired results may occur under complex maritime conditions, such as missed detections, false positives, and low detection accuracy. Moreover, the existing detection models endure large number of parameters and heavy computation cost. To deal with these problems, we suggest a lightweight ship model of detection called DSSM–LightNet based upon the improved YOLOv8n. First, we introduce a lightweight Dual Convolutional (DualConv) into the model to lower both the number of parameters and the computational complexity. The principle is that DualConv combines two types of convolution kernels, 3x3 and 1x1, and utilizes group convolution techniques to effectively reduce computational costs while processing the same input feature map channels. Second, we propose a Slim-neck structure in the neck network, which introduces GSConv and VoVGSCSP modules to construct an efficient feature-fusion layer. This fusion strategy helps the model better capture the features of targets of different sizes. Meanwhile, a spatially enhanced attention module (SEAM) is leveraged to integrate with a Feature Pyramid Network (FPN) and the Slim-neck to achieve simple yet effective feature extraction, minimizing information loss during feature fusion. CIoU may not accurately reflect the relative positional relationship between bounding boxes in some complex scenarios. In contrast, MPDIoU can provide more accurate positional information in bounding-box regression by directly minimizing point distance and considering comprehensive loss. Therefore, we utilize the minimum point distance IoU (MPDIoU) rather than the Complete Intersection over Union (CIoU) Loss to further enhance the detection precision of the suggested model. Comprehensive tests carried out on the publicly accessible SeaShips dataset have demonstrated that our model greatly exceeds other algorithms in relation to their detection accuracy and efficiency, while reserving its lightweight nature.
oceanography,engineering, marine, ocean