计算机工程与应用››2023,第59卷››发行(23): 165-174.内政部:10.3778/j.issn.1002-8331.2306-0205

图形图像处理 • 上一篇   下一篇

改进YOLOv8公司的道路损伤检测算法

李松,史涛,井方科  

  1. 1华北理工大学 电气工程学院,河北 唐山 063210
    2天津理工大学 电气工程与自动化学院,天津 300384
  • 出版日期:2023-12-01 发布日期:2023-12-01

改进的YOLOv8道路损伤检测算法

李松、石涛、荆方科  

  1. 1.华北科技大学电气工程学院,河北唐山063210
    2.天津理工大学电气工程与自动化学院,天津300384
  • 在线:2023-12-01 出版:2023-12-01

摘要:道路损伤检测是保障道路安全、实现道路损伤及时修复的一项重要任务。针对现有的道路损伤检测算法中检测效率低、成本高昂、难以应用于移动终端设备问题,提出了一种改进YOLOv8公司的轻量型道路损伤检测算法YOLOv8-道路损坏(YOLOv 8-RD)结合CNN和变压器的优势,提出了一种能够提取道路损伤图像全局特征信息和局部特征信息的机器人程序模块,以适应裂纹对象的大跨度与细长特征。在骨干网络末端和颈部网络中引入坐标注意力机制(协调注意力,CA)将位置信息嵌入到通道注意力中,强化特征提取能力,并抑制无关特征的干扰。在YOLOv8公司颈部网络中使用C2f幻影模块,以减少特征通道融合过程中的浮点运算量,降低模型参数量,同时提高特征表达性能。实验结果表明,在RDD2022和道路损坏数据集上,改进算法与原算法相比mAP50和3.7个百分点,而模型参数量仅为2.8×106,计算量仅为7.3×109,分别降低了6.7%和8.5%。算法检测速度达到88?FPS、能够实时准确检测道路损伤目标。通过与其他主流目标检测算法比较,验证了该方法的有效性和优越性。

关键词: 道路损伤检测, 深度学习, YOLOv8, 注意力机制, 变压器

摘要:道路损坏检测是保障道路安全、实现道路损坏及时修复的重要任务。针对现有道路损伤检测算法中检测效率低、成本高、难以应用于移动终端设备等问题,提出了一种基于改进YOLOV8的轻量级道路损伤检测方法YOLOV8-Road Damage(YOLOV 8-RD)。首先,结合CNN和Transformer的优点,提出了一种能够提取道路损伤图像全局和局部特征信息的BOT模块,以适应裂纹物体的大跨度和细长特征。然后,在主干网和颈部网络的末端引入协调注意(CA),将位置信息嵌入到信道注意中,增强特征提取能力,抑制无关特征的干扰。此外,在YOLOv8颈部网络中使用了C2fGhost模块,以减少特征通道融合过程中的浮点计算,减少模型参数的数量,提高特征表达性能。实验结果表明,在RDD2022数据集和道路损坏数据集中,与mAP50相比,改进算法分别提高了2%和3.7%,而模型参数数量仅为2.8×106,计算量仅为7.3×109,分别减少了6.7%和8.5%。该算法的检测速度达到88 FPS,能够实时准确地检测出道路损坏目标。通过与其他主流目标检测算法的比较,验证了该方法的有效性和优越性。

关键词: 道路损坏检测, 深度学习, YOLOv8, 注意机制, 变压器