High-Efficiency and High-Precision Ship Detection Algorithm Based on Improved YOLOv8n

Kun Lan,Xiaoliang Jiang,Xiaokang Ding,Huan Lin,Sixian Chan

DOI: https://doi.org/10.3390/math12071072

IF: 2.4

2024-04-03

Mathematics

Abstract:With the development of the intelligent vision industry, ship detection and identification technology has gradually become a research hotspot in the field of marine insurance and port logistics. However, due to the interference of rain, haze, waves, light, and other bad weather, the robustness and effectiveness of existing detection algorithms remain a continuous challenge. For this reason, an improved YOLOv8n algorithm is proposed for the detection of ship targets under unforeseen environmental conditions. In the proposed method, the efficient multi-scale attention module (C2f_EMAM) is introduced to integrate the context information of different scales so that the convolutional neural network can generate better pixel-level attention to high-level feature maps. In addition, a fully-concatenate bi-directional feature pyramid network (Concatenate_FBiFPN) is adopted to replace the simple superposition/addition of feature map, which can better solve the problem of feature propagation and information flow in target detection. An improved spatial pyramid pooling fast structure (SPPF2+1) is also designed to emphasize low-level pooling features and reduce the pooling depth to accommodate the information characteristics of the ship. A comparison experiment was conducted between other mainstream methods and our proposed algorithm. Results showed that our proposed algorithm outperformed other models by achieving 99.4% of accuracy, 98.2% of precision, 98.5% of recall, 99.1% of mAP@.5, and 85.4% of mAP@.5:.95 on the SeaShips dataset.

mathematics

What problem does this paper attempt to address?

### Problems Addressed by the Paper

This paper aims to address the challenges of ship detection in complex environments. Specifically, existing ship detection algorithms face challenges in robustness and effectiveness under adverse weather conditions such as rain, fog, waves, and lighting, as well as when dealing with factors like different sizes, scales, orientations, positions, and postures. Additionally, interference from background regions (such as floating objects, coastlines, and buildings) can also affect detection performance.

To tackle these issues, the authors propose an improved YOLOv8n network by introducing three advanced modules to enhance the accuracy and efficiency of ship detection:
1. **Efficient Multi-Scale Attention Module (C2f_EMAM)**: Integrates contextual information at different scales, significantly enhancing the attention to high-level feature maps.
2. **Fully Connected Bidirectional Feature Pyramid Network (Concatenate_FBiFPN)**: Optimizes the classification and regression structure, better addressing feature propagation and information flow issues in object detection.
3. **Spatial Pyramid Pooling Fast Structure (SPPF2+1)**: Redesigned to emphasize low-level pooling features and more comprehensively learn object features.

Experimental results show that the improved algorithm achieves 99.4% accuracy, 98.2% precision, 98.5% recall, 99.1% mAP@.5, and 85.4% mAP@.5:.95 on the SeaShips dataset, significantly outperforming other mainstream methods.