Abstract:Road damage detection is very important for road safety and timely repair. The previous detection methods mainly rely on humans or large machines, which are costly and inefficient. Existing algorithms are computationally expensive and difficult to arrange in edge detection devices. To solve this problem, we propose a lightweight and efficient road damage detection algorithm LE-YOLOv5 based on YOLOv5. We propose a global shuffle attention module to improve the shortcomings of the SE attention module in MobileNetV3, which in turn builds a better backbone feature extraction network. It greatly reduces the parameters and GFLOPS of the model while increasing the computational speed. To construct a simple and efficient neck network, a lightweight hybrid convolution is introduced into the neck network to replace the standard convolution. Meanwhile, we introduce the lightweight coordinate attention module into the cross-stage partial network module that was designed using the one-time aggregation method. Specifically, we propose a parameter-free attentional feature fusion (PAFF) module, which significantly enhances the model’s ability to capture contextual information at a long distance by guiding and enhancing correlation learning between the channel direction and spatial direction without introducing additional parameters. The K-means clustering algorithm is used to make the anchor boxes more suitable for the dataset. Finally, we use a label smoothing algorithm to improve the generalization ability of the model. The experimental results show that the LE-YOLOv5 proposed in this document can stably and effectively detect road damage. Compared to YOLOv5s, LE-YOLOv5 reduces the parameters by 52.6% and reduces the GFLOPS by 57.0%. However, notably, the mean average precision (mAP) of our model improves by 5.3%. This means that LE-YOLOv5 is much more lightweight while still providing excellent performance. We set up visualization experiments for multialgorithm comparative detection in a variety of complex road environments. The experimental results show that LE-YOLOv5 exhibits excellent robustness and reliability in complex road environments.

SEF-YOLO: an Improved YOLOv5 Algorithm for Traffic Sign Detection

Improved lightweight YOLOv5 based on ShuffleNet and its application on traffic signs detection

ETSR-YOLO: An improved multi-scale traffic sign detection algorithm based on YOLOv5

Efficient Vision Transformer YOLOv5 for Accurate and Fast Traffic Sign Detection

Traffic Sign Detection Based on the Improved YOLOv5

Small-target Traffic Sign Detection Based on Improved YOLOv8

An Improved Traffic Sign Detection and Recognition Deep Model Based on YOLOv5

Intelligent Recognition of Traffic Signs Based on Improved YOLO v3 Algorithm

LLTH-YOLOv5: A Real-Time Traffic Sign Detection Algorithm for Low-Light Scenes

Research on Traffic Sign Detection Algorithm Based on YOLO

TRD-YOLO: A Real-Time, High-Performance Small Traffic Sign Detection Algorithm

An Improved Light-Weight Traffic Sign Recognition Algorithm Based on YOLOv4-Tiny

YOLOv7-TS: A Traffic Sign Detection Model Based on Sub-Pixel Convolution and Feature Fusion

A real-time and lightweight traffic sign detection method based on ghost-YOLO

Traffic Sign Detection Algorithm Based on Improved YOLOv4-Tiny

Improved YOLO V5 with Balanced Feature Pyramid and Attention Module for Traffic Sign Detection

Improved YOLOv5-based for Small Traffic Sign Detection under Complex Weather

EDN-YOLO

YOLO-TSF: A Small Traffic Sign Detection Algorithm for Foggy Road Scenes

TLDM: An Enhanced Traffic Light Detection Model Based on YOLOv5

LE-YOLOv5: A Lightweight and Efficient Road Damage Detection Algorithm Based on Improved YOLOv5