MED-YOLOv8s: a new real-time road crack, pothole, and patch detection model

被引:20
作者
Zhao, Minghu [1 ]
Su, Yaoheng [1 ]
Wang, Jiuxin [1 ]
Liu, Xinru [2 ]
Wang, Kaihang [1 ]
Liu, Zishen [1 ]
Liu, Man [1 ]
Guo, Zhou [1 ]
机构
[1] Xian Polytech Univ, Sch Sci, Xian 710048, Peoples R China
[2] Xian Polytech Univ, Sch Elect Informat, Xian 710048, Peoples R China
关键词
Road damage; YOLOv8; Lightweight model; Data augmentation; Deep learning; DAMAGE DETECTION; AUTOMATIC DETECTION; CLASSIFICATION;
D O I
10.1007/s11554-023-01405-5
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Real-time road damage detection and assessment is crucial to ensure road safety. Traditional road damage detection methods mostly rely on manual labor, which is not only inefficient, but it is also difficult to guarantee its reliability. In this study, a road damage detection model, MED-YOLOv8s, based on YOLOv8s is proposed. MobileNetv3 is adopted as the backbone of the detection algorithm, which reduces the number of parameters and the number of computations in the process of feature extraction, enabling the model to achieve a good balance between the detection speed and the detection accuracy. The introduction of the ultralightweight attention mechanism, ECA, adapts the optimization of the correlation of channels to improve the model generalization performance. In addition, replacing the standard convolution with the DW convolution in the 21st layer of the network not only eliminates part of the redundant feature maps but also better extracts the correlation information between the feature maps. In this study, we also discuss the influence of the mix-up data augmentation weight parameter on the detection effect of the model. The experimental results show that the mAP@0.5 of the MED-YOLOv8s model proposed in this study is 95.2%, which is 1.1% higher than that of the original model, and at the same time, the calculation amount of the model is reduced by 46.2%. This method not only improves the detection accuracy but also greatly reduces the model complexity, providing a reference for subsequent model migration.
引用
收藏
页数:17
相关论文
共 64 条
[1]  
Alfarrarjeh A, 2018, IEEE INT CONF BIG DA, P5201, DOI 10.1109/BigData.2018.8621899
[2]  
Arya D., 2022, P 2022 IEEE INT C BI
[3]  
Bochkovskiy A, 2020, PREPRINT, DOI 10.48550/ARXIV.2004.10934
[4]   Road Damage Detection and Classification Using Mask R-CNN with DenseNet Backbone [J].
Chen, Qiqiang ;
Gan, Xinxin ;
Huang, Wei ;
Feng, Jingjing ;
Shim, H. .
CMC-COMPUTERS MATERIALS & CONTINUA, 2020, 65 (03) :2201-2215
[5]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[6]   Automatic recognition of surface cracks in bridges based on 2D-APES and mobile machine vision [J].
Dan, Danhui ;
Dan, Qiang .
MEASUREMENT, 2021, 168
[7]  
Ding W., 2022, P 2022 IEEE INT C BI
[8]   Modeling automatic pavement crack object detection and pixel-level segmentation [J].
Du, Yuchuan ;
Zhong, Shan ;
Fang, Hongyuan ;
Wang, Niannian ;
Liu, Chenglong ;
Wu, Difei ;
Sun, Yan ;
Xiang, Mang .
AUTOMATION IN CONSTRUCTION, 2023, 150
[9]  
Howard AG, 2017, Arxiv, DOI [arXiv:1704.04861, 10.48550/arXiv.1704.04861]
[10]  
Gao Y., 2023, Measurement, V219