EMG-YOLO: road crack detection algorithm for edge computing devices

被引:5
作者
Xing, Yan [1 ,2 ]
Han, Xu [1 ]
Pan, Xiaodong [3 ]
An, Dong [2 ]
Liu, Weidong [1 ]
Bai, Yuanshen [3 ]
机构
[1] Shenyang Jianzhu Univ, Sch Transportat & Surveying Engn, Shenyang, Liaoning, Peoples R China
[2] Shenyang Boyan Intelligent Transportat Technol Co, Shenyang, Liaoning, Peoples R China
[3] Shenyang Publ Secur Bur, Traff Police Div, Shenyang, Liaoning, Peoples R China
来源
FRONTIERS IN NEUROROBOTICS | 2024年 / 18卷
关键词
road crack detection; YOLOv5; Efficient Decoupled Head; MPDIou; deep learning;
D O I
10.3389/fnbot.2024.1423738
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Introduction Road cracks significantly shorten the service life of roads. Manual detection methods are inefficient and costly. The YOLOv5 model has made some progress in road crack detection. However, issues arise when deployed on edge computing devices. The main problem is that edge computing devices are directly connected to sensors. This results in the collection of noisy, poor-quality data. This problem adds computational burden to the model, potentially impacting its accuracy. To address these issues, this paper proposes a novel road crack detection algorithm named EMG-YOLO.Methods First, an Efficient Decoupled Header is introduced in YOLOv5 to optimize the head structure. This approach separates the classification task from the localization task. Each task can then focus on learning its most relevant features. This significantly reduces the model's computational resources and time. It also achieves faster convergence rates. Second, the IOU loss function in the model is upgraded to the MPDIOU loss function. This function works by minimizing the top-left and bottom-right point distances between the predicted bounding box and the actual labeled bounding box. The MPDIOU loss function addresses the complex computation and high computational burden of the current YOLOv5 model. Finally, the GCC3 module replaces the traditional convolution. It performs global context modeling with the input feature map to obtain global context information. This enhances the model's detection capabilities on edge computing devices.Results Experimental results show that the improved model has better performance in all parameter indicators compared to current mainstream algorithms. The EMG-YOLO model improves the accuracy of the YOLOv5 model by 2.7%. The mAP (0.5) and mAP (0.9) are improved by 2.9% and 0.9%, respectively. The new algorithm also outperforms the YOLOv5 model in complex environments on edge computing devices.Discussion The EMG-YOLO algorithm proposed in this paper effectively addresses the issues of poor data quality and high computational burden on edge computing devices. This is achieved through optimizing the model head structure, upgrading the loss function, and introducing global context modeling. Experimental results demonstrate significant improvements in both accuracy and efficiency, especially in complex environments. Future research can further optimize this algorithm and explore more lightweight and efficient object detection models for edge computing devices.
引用
收藏
页数:14
相关论文
共 28 条
  • [21] Transfer learning based bridge damage detection: Leveraging time-frequency features
    Talaei, Saeid
    Zhu, Xinqun
    Li, Jianchun
    Yu, Yang
    Chan, Tommy H. T.
    [J]. STRUCTURES, 2023, 57
  • [22] Tan MX, 2020, PROC CVPR IEEE, P10778, DOI 10.1109/CVPR42600.2020.01079
  • [23] Semantic Probability Distribution Modeling for Diverse Semantic Image Synthesis
    Tan, Zhentao
    Chu, Qi
    Chai, Menglei
    Chen, Dongdong
    Liao, Jing
    Liu, Qiankun
    Liu, Bin
    Hua, Gang
    Yu, Nenghai
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (05) : 6247 - 6264
  • [24] Crack-Detection Method for Asphalt Pavement Based on the Improved YOLOv5
    Tang, Gangting
    Yin, Chao
    Zhang, Xixuan
    Liu, Xinliang
    Li, Shufeng
    [J]. JOURNAL OF PERFORMANCE OF CONSTRUCTED FACILITIES, 2024, 38 (02)
  • [25] Wei Conghua, 2023, 2023 IEEE International Conference on Big Data (BigData), P1550, DOI 10.1109/BigData59044.2023.10386514
  • [26] A Page Object Detection Method Based on Mask R-CNN
    Xu, Canhui
    Shi, Cao
    Bi, Hengyue
    Liu, Chuanqi
    Yuan, Yongfeng
    Guo, Haoyan
    Chen, Yinong
    [J]. IEEE ACCESS, 2021, 9 : 143448 - 143457
  • [27] Zhang Wenwei, 2021, Advances in Neural Information Processing Systems, V34
  • [28] Zhang YM, 2021, Arxiv, DOI arXiv:2107.04829