A pavement distresses identification method optimized for YOLOv5s

被引:40
作者
Guo, Keyou [1 ]
He, Chengbo [1 ]
Yang, Min [1 ]
Wang, Sudong [1 ]
机构
[1] Beijing Technol & Business Univ, Beijing 100037, Peoples R China
关键词
D O I
10.1038/s41598-022-07527-3
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Automatic detection and recognition of pavement distresses is the key to timely repair of pavement. Repairing the pavement distresses in time can prevent the destruction of road structure and the occurrence of traffic accidents. However, some other factors, such as a single object category, shading and occlusion, make detection of pavement distresses very challenging. In order to solve these problems, we use the improved YOLOv5 model to detect various pavement distresses. We optimize the YOLOv5 model and introduce attention mechanism to enhance the robustness of the model. The improved model is more suitable for deployment in embedded devices. The optimized model is transplanted to the self-built intelligent mobile platform. Experimental results show that the improved network model proposed in this paper can effectively identify pavement distresses on the self-built intelligent mobile platform and datasets. The precision, recall and mAP are 95.5%, 94.3% and 95%. Compared with YOLOv5s and YOLOv4 models, the mAP of the improved YOLOv5s model is increased by 4.3% and 25.8%. This method can provide technical reference for pavement distresses detection robot.
引用
收藏
页数:15
相关论文
共 33 条
[1]  
[Anonymous], 2014, NEURAL INFORM PROCES
[2]  
Ba J. L., 2016, P ADV NEUR INF PROC
[3]  
Bochkovskiy A., 2020, ARXIV, DOI 10.48550/ARXIV.2004.10934
[4]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]   Feature selection for clustering - A filter solution [J].
Dash, M ;
Choi, K ;
Scheuermann, P ;
Liu, H .
2002 IEEE INTERNATIONAL CONFERENCE ON DATA MINING, PROCEEDINGS, 2002, :115-122
[7]   Sigmoid-weighted linear units for neural network function approximation in reinforcement learning [J].
Elfwing, Stefan ;
Uchibe, Eiji ;
Doya, Kenji .
NEURAL NETWORKS, 2018, 107 :3-11
[8]   Classification of EEG signals using the wavelet transform [J].
Hazarika, N ;
Chen, JZ ;
Tsoi, AC ;
Sergejew, A .
SIGNAL PROCESSING, 1997, 59 (01) :61-72
[9]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916
[10]   Digital Twin-Assisted Real-Time Traffic Data Prediction Method for 5G-Enabled Internet of Vehicles [J].
Hu, Chunhua ;
Fan, Weicun ;
Zeng, Elan ;
Hang, Zhi ;
Wang, Fan ;
Qi, Lianyong ;
Bhuiyan, Md Zakirul Alam .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2022, 18 (04) :2811-2819