YOLO-LRDD: a lightweight method for road damage detection based on improved YOLOv5s

Abstract

In computer vision, timely and accurate execution of object identification tasks is critical. However, present road damage detection approaches based on deep learning suffer from complex models and computationally time-consuming issues. To address these issues, we present a lightweight model for road damage identification by enhancing the YOLOv5s approach. The resulting algorithm, YOLO-LRDD, provides a good balance of detection precision and speed. First, we propose the novel backbone network Shuffle-ECANet by adding an ECA attention module into the lightweight model ShuffleNetV2. Second, to ensure reliable detection, we employ BiFPN rather than the original feature pyramid network since it improves the network's capacity to describe features. Moreover, in the model training phase, localization loss is modified to Focal-EIOU in order to get higher-quality anchor box. Lastly, we augment the well-known RDD2020 dataset with many samples of Chinese road scenes and compare YOLO-LRDD against several state-of-the-art object detection techniques. The smaller model of our YOLO-LRDD offers superior performance in terms of accuracy and efficiency, as determined by our experiments. Compared to YOLOv5s in particular, YOLO-LRDD improves single image recognition speed by 22.3% and reduces model size by 28.8% while maintaining comparable accuracy. In addition, it is easier to implant in mobile devices because its model is smaller and lighter than those of the other approaches.

Identification of illegal behaviors of sand trucks based on improved YOLO V5S

“…CA module refers to Channel Attention module, which is used to enhance the feature extraction capability of convolutional neural networks. The structure diagram of CA module is shown in the figure [3] . First, the input feature graph is globally average pooled in the width and height directions to obtain the feature graph in the width and height directions.…”

Efficient data dimensionality reduction method for improving road crack classification algorithms

“…Another recent work is Wan et al (2022). In it, a lightweight DL you only look once -lightweight method for road damage detection (YOLO-LRDD) is used and compared to Faster R-CNN.…”

Road damage detection with bounding box and generative adversarial networks based augmentation methods

“…In 2020, the researchers employed the U-Net architecture and the YOLOv3 algorithm to detect and classify various types of road damages in China [43]. Using a new backbone, Wan et al [44] presented the YOLO-LRDD algorithm. They increased the quantity of data in the RDD2020 dataset by using the data collected in China.…”

“…Using a new backbone, Wan et al. [44] presented the YOLO‐LRDD algorithm. They increased the quantity of data in the RDD2020 dataset by using the data collected in China.…”