关键词:
目标检测
深度学习
网络轻量化
剪枝算法
摘要:
随着深度学习技术的进步,车辆和行人道路目标检测算法虽然取得了显著进展,但其对计算资源的需求大,且显存消耗严重,导致在移动端和嵌入式设备等资源受限的场景中应用成本较高。轻量级神经网络以其较少的参数数量和较低的计算需求而著称,更适用于这些场景。针对当前检测模型参数多、计算量大的问题,本文从网络轻量化角度出发,研究了目标检测网络的重构方法、优化与剪枝技术,并探讨了轻量化网络在系统中的部署,旨在设计适用于中低端平台的轻量级算法。
本文所做的主要研究工作如下:
(1)轻量级目标检测网络重构方案研究。以YOLOV8n目标检测网络为研究对象,将YOLOV8n模型中的backbone主干网络替换成更为轻量级的Star Net网络结构,改进Bottleneck创新重构了SC2f模块,优化不同尺度特征的融合效果,并利用共享卷积的思想设计高效轻量的检测头(Lightweight Detection Head,LHead),以弥补轻量化网络改进的精度损失,构建一个新的轻量化深度学习目标检测网络YOLOV8-S-S-L。通过实验验证,改进后的算法能够保证模型检测平均精度(Mean Average Precision,MAP)为原模型98%的前提下,将模型的大小和参数量减少了45%,计算量降低了55%。此外,热力图结果显示,该方法与原模型在相同图像上关注区域基本相同。
(2)深度学习目标检测网络优化及剪枝技术研究。针对交通道路场景特性,结合有效的特征融合模块(Scale Sequence Feature Fusion,SSFF)进行优化以及在原有的多尺度特征融合网络结构中增加小目标检测层,从而提升YOLOV8n目标检测网络的识别精度,构建高精度的YOLOV8n-S-P网络,以避免剪枝操作导致识别精度大幅下降影响整体性能。最终,为了降低模型对中低端硬件计算能力的要求,本研究采用了基于幅值的层级自适应稀疏剪枝技术,成功开发出新型的YOLOV8-S-P-L轻量级网络架构。实验结果表明,新构建的网络参数量减少了67%、计算量下降了35%的情况下,仍然提升了2.4%的精度,在实现模型轻量化的同时,显著提升了网络在交通道路环境中对车辆目标的检测效能。
(3)道路目标检测系统的设计与实现。结合本文所提出算法设计了基于Py Qt5和Open CV的道路目标检测系统,可以对行人车辆的图像或者视频进行目标检测。实验结果表明,在有效降低深度学习网络参数量、运算量及模型大小的同时该系统可以准确的对行人车辆图像或者视频进行检测,并实时将检测结果及统计数量显示在系统界面。
本文提出的面向交通道路场景下轻量化目标检测算法,通过优化模型的计算和存储效率,在保持较高精度和实时性能的基础上,成功实现了模型的轻量化。这一改进确保了即使在复杂的道路环境中,模型依然能够表现出较强的稳定性和可靠性。研究成果不仅具有重要的理论价值,也对智能交通和自动驾驶技术的发展具有积极的推动作用。