Flexi-YOLO: A lightweight method for road crack detection in complex environments

Road crack detection is critical to global infrastructure maintenance and public safety, and complex background environments and nonlinear damage crack patterns challenge the need for real-time, efficient, and accurate detection.This paper proposes a lightweight yet robust Flexi-YOLO model based on the YOLOv8 algorithm. We designed Wise-IoU as the model's loss function to optimize the regression accuracy of its bounding boxes and enhance robustness to low-quality samples. The DCNv-C2f module is constructed for the transformation and fusion of feature information, allowing the convolutional kernels to adapt to the complex shape characteristics of cracks dynamically. A Global Attention Module (GAM) is integrated to improve the model's perception of global information. The AKConv convolution operation is employed to adaptively adjust the size of convolutions, further enhancing local feature capturing. Additionally, a lightweight network design is implemented, establishing G-Head (Ghost-Head) as the detection head to optimize the issue of feature redundancy. Experimental results show that Flexi-YOLO achieves an accuracy increase of 2.7% over YOLOv8n, a recall rate rise of 4.7%, a mAP improvement of 5.3%, a [email protected] increase of 3.9%, a decrease of 0.5 in GFLOPS, and an F1 score improvement from 0.80 to 0.84. Flexi-YOLO offers higher detection accuracy and robustness and meets the industrial demands for lightweight real-time detection and lower application costs, providing an efficient and precise solution for the automated detection of road cracks.