《计算机应用》唯一官方网站

• •    下一篇

基于改进YOLOv8的嵌入式道路裂缝检测算法

耿焕同1,刘振宇2,蒋骏2,范子辰3,李嘉兴2   

  1. 1. 南京信息工程大学 计算机与软件学院, 南京 210044
    2. 南京信息工程大学计算机学院
    3. 南京信息工程大学软件学院
  • 收稿日期:2023-05-23 修回日期:2023-08-16 发布日期:2023-09-01 出版日期:2023-09-01
  • 通讯作者: 刘振宇
  • 基金资助:
    国家自然科学基金

Embedded road crack detection algorithm based on improved YOLOv8

  • Received:2023-05-23 Revised:2023-08-16 Online:2023-09-01 Published:2023-09-01
  • Supported by:
    the National Natural Science Foundation of China

摘要: 在边缘端设备部署YOLOv8 l模型进行道路裂缝检测可以实现较高的精度,但难以保证实时检测。针对此问题,提出一种可部署到边缘计算设备Jetson AGX Xavier上的基于改进YOLOv8模型的目标检测算法。首先,使用部分卷积设计了Faster Block结构,用以替换YOLOv8 C2f模块中的Bottleneck结构,并将改进后的C2f模块记为C2f-Faster;其次,在YOLOv8主干网络中的每个C2f-Faster模块之后接一个SE(Squeeze-and Excitation)通道注意力层,进一步提高检测的精度。在开源道路损害数据集RDD20(Road Damage Detection 20)上实验的结果表明:所提方法的平均F1得分为0.573,每秒检测帧数(Frames per Second,FPS)为47,模型大小为55.5MB,相较于GRDDC20(Global Road Damage Detection Challenge20)的SOTA(state-of-the-art )模型,在F1得分上提高了0.8个百分点,在FPS上提高了292%,同时减小了42%的模型大小,实现了在边缘设备上对道路裂缝实时且准确的检测。

关键词: YOLOv8, 目标检测, 轻量化, 注意力机制, 道路裂缝

Abstract: Deploying the YOLOv8 l model on the edge device for road crack detection can achieve high accuracy, but it is difficult to guarantee real-time detection. To solve this problem, a target detection algorithm based on the improved YOLOv8 model that can be deployed on the edge computing device Jetson AGX Xavier is proposed. First, the Faster Block structure was designed using partial convolution to replace the Bottleneck structure in the YOLOv8 C2f module, and the improved C2f module was recorded as C2f-Faster; Secondly, an SE (Squeeze-and Excitation) channel attention layer was connected after each C2f-Faster module in the YOLOv8 backbone network to further improve the detection accuracy. The experimental results on the open source road damage dataset RDD20 (Road Damage Detection 20) show that the average F1 score of the proposed method is 0.573, the number of detection frames per second (FPS) is 47, and the model size is 55.5MB. Compared with the SOTA (state-of-the-art) model of GRDDC20 (Global Road Damage Detection Challenge20), the F1 score has increased by 0.8 percentage points, the FPS has increased by 292%, and the model has been reduced by 42%. It realizes real-time and accurate detection of road cracks on edge devices.

Key words: YOLOv8(You Only Look Once version 8), object detection, lightweight, attention mechanism, road crack

中图分类号: