关键词:
交通标志
YOLO
目标检测
注意力机制
摘要:
在社会经济迅速发展和汽车数量大幅增加的背景下,道路交通安全问题愈发突出。无人驾驶技术和智能交通系统成为应对这一挑战的重要方向。交通标志作为道路信息的关键载体,对驾驶安全至关重要。然而,在实际驾驶中交通标志可能因各种原因被忽略,增加事故风险。同时,传统交通标志识别方法在复杂场景下存在精度不足和计算效率低下的问题。若系统识别不准确,会向无人驾驶车辆传递错误信息,导致行车决策失误。此外,车载设备和嵌入式系统计算资源有限,自动驾驶车辆需实时运行,模型复杂可能导致识别延迟。因此,开发一种高效、准确且轻量化的交通标志识别算法至关重要,它不仅能提升道路安全性和交通效率,还能推动无人驾驶和智能交通系统的发展,对于确保道路交通安全和提升通行效能具有重要的现实意义。本文基于YOLOv11的轻量化交通标志识别算法进行了研究与设计,主要研究内容如下:
(1)在深度学习的交通标志识别领域,国内公开的数据集样本不足是一个挑战。国内目前仅有的公开交通标志数据集,难以满足日益复杂的交通标志识别需求。为解决这一问题,本文展开了深入研究,整合了一个交通标志数据集TTKTS。并且,为了有效扩充数据集并提高模型的泛化能力,本研究设计并实现了一个图像增强处理工具。为后续交通标志识别模型的训练提供了更充足、更优质的数据支持,为有效提升交通标志识别系统的性能与泛化能力奠定了基础。
(2)基于Faster Net和组卷积增强的轻量化识别算法。卷积神经网络在训练过程中通常面临耗时长、模型规模庞大以及内存占用高的问题,这些问题限制了其在交通标志识别场景下的轻量化部署。本文提出了一种针对YOLOv11算法模型的轻量化改进方案。通过构建并引入融合网络、优化检测头对模型网络结构进行改进,在确保识别准确率保持在较高水平的同时,显著降低计算量与参数量,从而满足轻量化部署的需求。
(3)融合CAGA注意力机制和CARAFE上采样的改进算法。尽管轻量化模型设计在一定程度上降低了模型复杂度,但同时也带来了识别准确率下降、误检与漏检现象频发等问题,尤其是在处理图像中的小目标时,识别效果欠佳,且识别速度难以满足交通标志识别的实时性要求。本文针对改进后的基于Faster Net和组卷积增强的轻量化识别算法,提出了一种精度与速度优化方案。该方案对网络结构中的上采样方法和损失函数进行了优化,并创新性地引入一种结合空间注意力和通道注意力的注意力机制,有效提升了目标识别的准确率与推理速度,进而更好地满足自动驾驶场景下对实时性的严格要求。
(4)基于改进YOLOv11的交通标志识别系统。为了更直观地呈现模型的实际效果,本文采用Py Qt5框架进行系统开发,并完成了模型的本地部署。在系统设计中,共构建了10个功能模块,其中包括标注工具、权重选择、参数调整、本地图片和视频检测、实时摄像头检测、识别、输出打印、结果展示以及存储结果等,旨在提供一个全面而高效的交通标志检识别解决方案。
图[36]表[15]参[72]