Road underground defect detection in ground penetrating radar images based on an improved YOLOv5s model

• Published in: Journal of applied geophysics Vol. 229; p. 105491
• Main Authors: Xue, Wei, Li, Ting, Peng, Jiao, Liu, Li, Zhang, Jian
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 01.10.2024
• Subjects: Attention mechanism; DenseNet; Focal loss function; Ground penetrating radar; Road underground defect detection; YOLOv5s; DenseNet; Attention mechanism; Ground penetrating radar; YOLOv5s; Road underground defect detection; Focal loss function
• ISSN: 0926-9851
• DOI: 10.1016/j.jappgeo.2024.105491

Road underground defect detection plays a crucial role in assessing transportation infrastructure. Ground penetrating radar (GPR) serves as a widely used geophysical tool for this purpose. However, the traditional manual interpretation of GPR images heavily relies on the experience of the practitioner, leading to inefficiency and inaccuracies. To tackle these challenges, this paper proposes an automatic detection method for underground defects of roads based on an improved YOLOv5s model. First, the dense connection structure is integrated in the C3 module of the backbone to form the Dense-C3 module to enhance the capability of feature extraction. Subsequently, a convolutional block attention module (CBAM) is incorporated after each Dense-C3 module to refine features and enhance efficiency. Furthermore, the focal loss function is employed for the confidence loss to mitigate the impact of sample imbalance on detection performance. Experimental results demonstrate that the proposed model achieves a mean average precision (mAP) of 96.4% for synthetic data and 91.9% for real data, outperforming seven other models. The detection speed of the proposed model for real data reaches 51 frames per second, meeting the real-time detection requirements of road underground defects. •The Dense-C3 module is constructed to improve the feature extraction ability.•The CBAM is added after each Dense-C3 module to refine features.•The focal loss function is used to mitigate the impact of sample imbalance.