[发明专利]一种普适性轻量级深度学习车辆检测方法

说明书

本发明提出了一种普适性轻量级深度学习车辆检测方法，用于解决现有技术中在计算资源、存储空间有限的情况下，无法同时满足准确率和实时性的问题，实现步骤为：获取训练样本集和测试样本集；构建改进的YOLOv4网络模型；对改进的YOLOv4网络模型进行迭代训练；获取车辆检测结果。本发明所构建的主干特征提取网络采用深度可分离卷积模块代替传统卷积模块方式，降低模型大小的同时提高检测实时性，同时所构建的主干特征提取网络所包含的第二瓶颈模块中加入通道混洗模块，一定程度上提高了车辆检测的精度，可用于汽车自动或半自动驾驶。

技术领域

本发明属于深度学习技术领域，特别涉及一种普适性轻量级深度学习车辆检测方法。考虑现实场景中车载终端计算资源、存储空间有限而导致大多数深度学习方法不再适用，根据轻量化网络模型的思想，通过对YOLOv4网络模型中主干特征提取网络轻量化设计，在保证了检测的准确率的前提下，降低了车辆检测复杂度、压缩了模型大小，提高了检测的实时性，可用于高级辅助驾驶系统中。

背景技术

研究表明，目前我国智能网联汽车的发展正处在从L2级自动驾驶向L3级自动驾驶过度阶段。然而随着道路上的车辆日益增多，汽车在驾驶过程中容易发生各种突发状况，给人们带来经济损失的同时，可能造成部分区域的交通拥堵，更加严重的是可能导致相关人员的伤亡。随着智能交通系统的蓬勃发展能够有针对性的部分解决当前所面临的各种各样的交通问题，其中高级辅助驾驶系统(Advanced Driving Assistance System，ADAS)作为智能交通系统体系框架的重要内容，可以有针对性的解决道路上各种问题。高精度、实时车辆检测则是自动驾驶和ADAS的基础，同时也是ADAS中至关重要的一步。大多数车辆检测算法的提出都将提高精度作为第一准则，但不可避免的是模型复杂度的提升和大量内存的消耗。在现实场景中，由于单车计算资源、存储空间有限，很难同时满足高准确率和实时性，因此如何在有限的计算资源和存储资源的场景下保证准确率的同时提高检测的实时性是研究的关键。

现有基于深度学习的车辆检测方法有两种，一种是两阶段车辆检测方法，另一种是单阶段车辆检测方法。采用两阶段车辆检测主要通过一个完整的卷积神经网络(Convolutional Neural Networks，CNN)来完成目标检测的过程。在训练网络时，主要训练两个部分，第一步是训练候选区生成网络(Region Proposal Network，RPN)，第二步则是训练目标区域检测的网络。该方法虽然具有很高的检测精度，但由于网络结构的复杂性，导致检测速度很慢，往往在一些场景下无法满足实时性的要求。采用单阶段车辆检测主要是通过直接回归的方式进行车辆检测，该方法虽然在检测速度上相较于两阶段算法有较大的优势，但在计算资源和存储资源有限的场景下仍无法满足实时性，为了进一步提高检测实时性，需要对单阶段车辆检测进行轻量化操作，其中包括剪枝和构建轻量级网络模型，但由于单阶段检测网络结构比较简单，对图像中车辆信息提取能力有限，从而导致检测精度较低。

现有的深度车辆检测算法，多数采用YOLOv3网络模型为基础进行改进来提高检测速度，例如申请公布号为CN 110796168 A，名称为“一种基于改进YOLOv3的车辆检测方法”的专利申请，公开的一种改进YOLOv3的车辆检测方法，该方法重新设计Darknet层于三个yolo层之间的卷积神经网络结构，借鉴了TridentNet权重共享的想法，设计YOLO-TN网络；对YOLO-TN卷积神经网络进行模型剪枝；构建车辆检测数据集，并对数据集中的车辆位置信息进行标注；训练YOLO-TN的车辆检测模型，完成车辆检测任务。该方法在保证实时性的同时还具有较高的平均精度，对远处车辆以及小尺度目标进行检测时，漏检率较低，定位也更准确。但其存在的不足之处在于，该方法采用模型剪枝的方法对网络模型进行轻量化，在计算资源有限和存储资源有限的场景下无法同时保证高精度和实时性。