关键词:
自动驾驶技术
目标检测算法
复杂道路场景
小目标检测
算法轻量化
摘要:
随着汽车保有量的快速增长,城市道路拥堵和交通事故伤亡等问题日益严重。在此背景下,自动驾驶技术作为解决传统道路交通问题的重要途径,受到了广泛关注。目标检测算法作为自动驾驶系统中环境感知的基础,需快速准确地识别并定位目标。然而,当车辆行驶在复杂道路场景中时,由于目标数量繁多且分布杂乱,同时距离和角度变化幅度较大,容易出现密集目标遮挡和小目标检测困难等问题,从而导致误检与漏检的发生,对目标检测算法提出了巨大挑战。同时,自动驾驶系统复杂度的不断提升也给硬件带来沉重负担。因此,针对复杂道路场景下的目标检测算法展开研究具有重大现实意义。本研究以YOLOv8n深度学习目标检测算法为基础,重点聚焦复杂道路场景中面临的小目标检测问题以及算法轻量化需求,深入探索有效的目标检测算法,以提高自动驾驶系统在复杂道路环境下的性能和可靠性。具体研究内容如下:
(1)针对道路多目标检测中存在的遮挡目标及小目标检测困难的问题,提出了一种基于YOLOv8n的改进算法RSFM-YOLO。该结构的主要改进在于引入了变压器解码器头,该解码器头可优化地面实况与预测方框之间的匹配,从而有效解决物体重叠和多尺度检测问题。此外,还加入了小目标检测层,以保留有利于检测小目标的关键信息,从而显著提高对小目标的检测精度。为了提高训练时的学习能力和减少冗余计算,采用Faster Net主干网络来替代CSPDarknet53,从而加快训练过程。最后,引入INNER-MPDIo U损失函数来替代原算法的Clo U损失函数,以加速收敛并获得更精确的回归结果。本文在不同的数据集上进行了一系列实验,实验结果表明,在精度检测方面,所提出的模型RSFM-YOLO优于原始模型YOLOv8n。在Vis Drone、Tiny Person和VSCrowd数据集上,所提出的方法平均准确率分别提高了7.9%、12.3%和4.5%,有效提升了对复杂道路中车辆和行人目标的检测性能。
(2)针对移动驾驶平台计算能力有限的问题,提出了一种基于YOLOv8n改进的轻量化算法LBT-YOLO。首先,用线性可变形卷积替换了部分传统卷积层,并设计了C2L模块,从而在减少模型参数量的同时保持了检测精度;其次,在加权双向特征金字塔网络的基础上,设计了一种新的颈部网络结构-双向定位特征金字塔网络,增强了特征融合和上下文信息的交互,提高了模型的检测精度;最后,提出了一种新的检测头-任务对齐动态检测头。该检测头通过共享颈部网络特征减少参数量,并进行任务分解对齐,利用动态卷积和动态特征选择实现高精度目标检测。经过一系列改进后,LBT-YOLO在自动驾驶BDD100K数据集上的表现优于YOLOv8n,模型参数量减少了48.2%,计算量降低了0.613G,同时平均准确率提高了2.4%,实现了算法的轻量化。
(3)基于PyQt5设计道路目标检测系统用来部署本文的改进算法。该系统可以加载图片视频进行检测,也可用直接用摄像头直接进行检测,并且支持同时加载两种不同的模型对图片或视频进行直观的检测效果对比。与此同时依托现实道路场景自制数据集对改进算法进行验证,RSFM-YOLO和LBT-YOLO相较于基准模型的平均准确率分别提高了7.7%、1.3%,有效验证了改进算法在复杂道路场景下的鲁棒性与泛化能力。