Research on YOLOv5s Improved Algorithm for Pavement Crack Detection in Complex Environments

• Published in: IEEE access Vol. 12; pp. 122452 - 122461
• Main Authors: Wu, Xiao, Ma, Tao, Zhao, Qipeng, Zhu, Liucun, He, Congwei
• Format: Journal Article
• Language: English
• Published: IEEE 2024
• Subjects: Accuracy; Complex environments; Deep learning; Defect detection; Feature extraction; pavement crack detection; Roads; Surface cracks; YOLO; YOLOv5s
• ISSN: 2169-3536; 2169-3536
• DOI: 10.1109/ACCESS.2024.3451708

Aiming to solve the problem of pavement crack detection in complex road environments, an improved algorithm based on YOLOv5s is proposed. First, the CBAM (Convolutional Block Attention Module) is introduced after the backbone network's C3 modules to focus the complex scene's practical information and enhance the model's attention to the crack region. Second, the BiFPN (Bidirectional Feature Pyramid Network) is used in Neck to replace the bidirectional PANet (Path Aggregation Network) in YOLOv5s to improve the multiscale feature fusion, which reduces detection leakage due to illumination and scale factors. Thirdly, the P6 detection head for denser cracks is added to the Head, and the CA (Coordinate Attention) module is introduced to improve the crack detection capability at multiple scales. Finally, the algorithm is experimentally compared with SSD (Single Shot MultiBox Detector), Faster-RCNN (Faster-Region Convolutional Neural Network), and unimproved YOLOv5s on the constructed pavement crack image dataset. The results show that compared with other algorithms, when mAP@0.5 (mean Average Precision when IoU=0.5) values are improved by 10.6%, 9.2%, and 4.6%, respectively, and cracks are identified better than the other three models.