计算机工程与应用››2024,第60卷››发行(8): 202-212.内政部:10.3778/j.issn.1002-8331.2309-0415

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

改进YOLOv8公司的道路交通标志目标检测算法

田鹏,毛力  

  1. 江南大学 人工智能与计算机学院,江苏 无锡 214122
  • 出版日期:2024-04-15 发布日期:2024-04-15

改进的YOLOv8交通标志目标检测算法

田鹏、毛利  

  1. 江南大学人工智能与计算机科学学院,江苏无锡214122
  • 在线:2024-04-15 出版:2024-04-15

摘要:虽趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋趋YOLOv8公司改进的道路交通标志目标检测算法。由于小目标检测中容易出现漏检的现象,利用BRA(双向路由注意)注意力机制提高网络对小目标的感知能力。此外,还利用可形变卷积模块DCNv3(可变形卷积v3)针对特征图中的不规则形状具有更好的特征提取能力,使骨干网络更好地适应不规则的空间结构,更精准地关注重要目标,从而提高模型对遮挡重叠目标的检测能力。DCNv3和BRA模块均在基本不增加模型权重大小的情况下提高模型准确性。同时引入基于辅助边框的内部IOU路标、CCTSDB、TSDD、GTSDB四个数据集上,分别进行了小样本训练、大样本训练、单目标检测和多目标检测,实验结果均有所提高。其中,在道路标志数据集上的实验结果最佳,YOLOv8公司改进模型的均值平均精度mAP50:95分别为90.7%和75.1%相较于基线模型,mAP50:95 5.9和4.8个百分点。实验结果表明,YOLOv8公司改进模型有效地实现了在复杂道路场景下的交通标志检测。

关键词: YOLOv8, 小目标检测, 注意力机制, 复杂道路场景

摘要:虽然目前的测试技术日趋成熟,但复杂环境下的小目标检测仍是研究的难点。针对道路交通场景中交通标志目标比例高、小目标目标比例高和环境干扰因素大的问题,提出了一种基于YOLOv8改进的道路交通标志目标测试算法。针对小目标测试中容易出现漏检的问题,采用了双层路由注意(BRA)机制来提高网络对小目标的感知能力。此外,它还使用了形状变换卷积模块可变形卷积V3(DCNV3)。它对特征图中的不规则形状具有更好的特征提取能力,使骨干网络能够更好地适应不规则的空间结构,并更准确地关注重要的注意力、目标,从而提高模型对重叠目标的检测能力。DCNV3和BRA模块都在不增加模型权重的情况下提高了模型的准确性。同时,引入了基于辅助边界的Inner-IOU损失函数。在RoadSign、CCTSDB、TSDD和GTSDB四个数据集上,进行了小样本训练、大样本训练、单目标检测和多目标检测。实验结果得到了改进。其中,对路标数据集的实验是最好的。改进的YOLOv8模型mAP50和mAP50:95的平均准确度分别达到90.7%和75.1%。与基线模型相比,mAP50和mAP50:95分别增加了5.9和4.8个百分点。实验结果表明,改进的YOLOV8模型有效地实现了复杂道路场景中的交通符号检测。

关键词: YOLOv8, 小目标检测, 可变形卷积, 注意机制, 复杂的道路场景