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 条
  • [1] Road damage detection with bounding box and generative adversarial networks based augmentation methods
    Aghayan-Mashhady, Nima
    Amirkhani, Abdollah
    [J]. IET IMAGE PROCESSING, 2024, 18 (01) : 154 - 174
  • [2] A hybrid deep learning pavement crack semantic segmentation
    Al-Huda, Zaid
    Peng, Bo
    Algburi, Riyadh Nazar Ali
    Al-antari, Mugahed A.
    AL-Jarazi, Rabea
    Zhai, Donghai
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 122
  • [3] Dynamic Head: Unifying Object Detection Heads with Attentions
    Dai, Xiyang
    Chen, Yinpeng
    Xiao, Bin
    Chen, Dongdong
    Liu, Mengchen
    Yuan, Lu
    Zhang, Lei
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 7369 - 7378
  • [4] CenterNet: Keypoint Triplets for Object Detection
    Duan, Kaiwen
    Bai, Song
    Xie, Lingxi
    Qi, Honggang
    Huang, Qingming
    Tian, Qi
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 6568 - 6577
  • [5] YOLO-ReT: Towards High Accuracy Real-time Object Detection on Edge GPUs
    Ganesh, Prakhar
    Chen, Yao
    Yang, Yin
    Chen, Deming
    Winslett, Marianne
    [J]. 2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 1311 - 1321
  • [6] Road damage detection algorithm for improved YOLOv5
    Guo, Gege
    Zhang, Zhenyu
    [J]. SCIENTIFIC REPORTS, 2022, 12 (01)
  • [7] A Deep Learning Method for Pavement Crack Identification Based on Limited Field Images
    Hou, Yue
    Liu, Shuo
    Cao, Dandan
    Peng, Bo
    Liu, Zhuo
    Sun, Wenjuan
    Chen, Ning
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (11) : 22156 - 22165
  • [8] 3D vision technologies for a self-developed structural external crack damage recognition robot
    Hu, Kewei
    Chen, Zheng
    Kang, Hanwen
    Tang, Yunchao
    [J]. AUTOMATION IN CONSTRUCTION, 2024, 159
  • [9] RDD-YOLOv5: Road Defect Detection Algorithm with Self-Attention Based on Unmanned Aerial Vehicle Inspection
    Jiang, Yutian
    Yan, Haotian
    Zhang, Yiru
    Wu, Keqiang
    Liu, Ruiyuan
    Lin, Ciyun
    [J]. SENSORS, 2023, 23 (19)
  • [10] Edge YOLO: Real-Time Intelligent Object Detection System Based on Edge-Cloud Cooperation in Autonomous Vehicles
    Liang, Siyuan
    Wu, Hao
    Zhen, Li
    Hua, Qiaozhi
    Garg, Sahil
    Kaddoum, Georges
    Hassan, Mohammad Mehedi
    Yu, Keping
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (12) : 25345 - 25360