YOLOv8-PD: an improved road damage detection algorithm based on YOLOv8n model

Jiayi Zeng,Han Zhong

DOI: https://doi.org/10.1038/s41598-024-62933-z

IF: 4.6

2024-05-29

Scientific Reports

Abstract:Road damage detection is an crucial task to ensure road safety. To tackle the issues of poor performance on multi-scale pavement distresses and high costs in detection task, this paper presents an improved lightweight road damage detection algorithm based on YOLOv8n, named YOLOv8-PD (pavement distress). Firstly, a BOT module that can extract global information of road damage images is proposed to adapt to the large-span features of crack objects. Secondly, the introduction of the large separable kernel attention (LKSA) mechanism enhances the detection accuracy of the algorithm. Then, a C2fGhost block is constructed in the neck network to strengthen the feature extraction of complex road damages while reducing the computational load. Furthermore, we introduced lightweight shared convolution detection head (LSCD-Head) to improve feature expressiveness and reduce the number of parameters. Finally, extensive experiments on the RDD2022 dataset yield a model with parametric and computational quantities of 2.3M and 6.1 GFLOPs, which are only 74.1% and 74.3% of the baseline, and the mAP reaches an improvement of 1.4 percentage points from the baseline. In addition, experimental results on the RoadDamage dataset show that the mAP increased by 4.2% and this algorithm has good robustness. This method can provide a reference for the automatic detection method of pavement distress.

multidisciplinary sciences

What problem does this paper attempt to address?

The paper aims to solve the efficiency and accuracy issues in road damage detection. Traditional methods such as manual inspection and deep learning-based algorithms face challenges of insufficient performance, high cost, and slow speed in detecting road surface damages like cracks. To address this, researchers propose an improved lightweight algorithm called YOLOv8-PD, based on the YOLOv8n model. This algorithm enhances performance in the following ways:

1. Introducing a new structure called BOT module to adapt to large-scale feature extraction of crack objects.
2. Incorporating Large Kernel Separable Attention (LKSA) mechanism to enhance detection accuracy.
3. Constructing C2fGhost block to strengthen feature extraction for complex road damages while reducing computational load.
4. Utilizing a lightweight shared convolutional detection head (LSCD-Head) to improve feature representation capability and reduce parameter count.

Experimental results show that YOLOv8-PD reduces parameters and computational complexity by approximately 26% and 25% respectively on the RDD2022 dataset, while improving mean average precision (mAP) by 1.4 percentage points. Furthermore, it achieves a 4.2% increase in mAP on the RoadDamage dataset, demonstrating good robustness. This method provides an effective reference for automatic detection of road damages.