QL-YOLOv8s: Precisely Optimized Lightweight YOLOv8 Pavement Disease Detection Model

被引:1
作者
Guo, Jinbo [1 ]
Wang, Shenghuai [1 ]
Chen, Xiaohui [1 ]
Wang, Chen [1 ]
Zhang, Wei [1 ]
机构
[1] Hubei Univ Automot Technol, Sch Mech Engn, Shiyan 442002, Peoples R China
来源
IEEE ACCESS | 2024年 / 12卷
基金
中国国家自然科学基金;
关键词
Accuracy; YOLO; Feature extraction; Road traffic; Object recognition; Surface cracks; Defect detection; Road transportation; Maintenance engineering; Road surface disease detection; lightweight; YOLOv8; MLCA; DWR; BiFPN; NETWORK;
D O I
10.1109/ACCESS.2024.3452129
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Detecting road surface defects is essential for highway maintenance, yet the application of most models is hindered by the limitations of existing detection resources. To address this challenge, we have enhanced YOLOv8, introducing a lightweight detection model dubbed QL-YOLOv8s. In this study, we employ the DIoU loss function to optimize bounding box regression, taking into account both the size of overlapping areas and the distance between the centers of boxes, thereby handling targets of various sizes and shapes with improved localization accuracy. Moreover, a lightweight Mixed Local Channel Attention (MLCA) has been incorporated into the backbone of the model, aimed at enhancing the recognition capabilities in complex environments without in-creasing the model's burden. Furthermore, by integrating the Dilated Wrapping Residual (DWR) module and C2f into BiFPN, we developed a new neck structure, BiFPN-D, and introduced a lightweight detection head, Detect-T3, thus augmenting the model's feature perception capacity, reducing parameter count, and boosting detection speed. Based on the RDD 2022 public dataset, QL-YOLOv8s demonstrated a reduction in parameter count and size by 37%, a decrease in com-putational requirements by 19%, and achieved an average precision of mAP0.5 at 95.8%. These results underscore the contribution and practical value of our method to the technology of automatic road defect detection.
引用
收藏
页码:128392 / 128403
页数:12
相关论文
empty
未找到相关数据