[发明专利]卡口抓拍图像车辆检测方法、计算机存储介质及电子设备

说明书

本发明提供了一种卡口抓拍图像车辆检测方法、计算机存储介质及电子设备，方法包括以下步骤：S1、获取卡口抓拍的图像；S2、对卡口抓拍的图像进行识别，并采用车身检测模型对识别到的车辆进行车身检测以检测出车身区域；S3、建立过滤规则，筛选并过滤掉不完整车身的图像信息，得到完整的车身区域；S4、采用车身前后拍分类模型对完整的车身区域进行分类；S5、输出完整的车辆信息。根据本发明实施例的卡口抓拍图像车辆检测方法，通过对现有的车辆检测方案进行优化，改变SSD中检测和分类同时进行的端到端的做法，将检测和分类分步完成，即将原有的检测分类同时进行改为先检测，再分类，可大幅提高车辆的检出率，精确率并实现模型的轻量化。

技术领域

本发明涉及车辆检测领域，更具体地，涉及一种卡口抓拍图像车辆检测方法、计算机存储介质及电子设备。

背景技术

卡口抓拍图像中的车辆检测是车辆视频监控系统的关键环节，利用检测装置在图像中快速准确地检测出车辆目标，对后续的车辆属性分析和识别非常重要，也极大地影响着系统的处理性能。检出率即图像中被检出的目标数占全部待检目标总数的比例，这个指标越接近1，代表着越多的目标被检出，极少的目标被漏检，意味着车辆视频监控系统输出的结果更真实可靠。因此，一个具有高检出率和高精确率的车辆检测方案对于完善和增强道路车辆视频监控系统的功能和性能至关重要。近年来，随着计算机算力的提高和深度学习理论与方法的推广应用，涌现出一大批基于深度神经网络的端到端的目标检测模型，例如YOLO、SSD、MobileNetSSD、TinyDSOD等。其中，YOLO算法的精确率在众多算法中居于首位，但是其计算速度太慢，难以在工程中应用；SSD及其改进算法因其较高的检测精度和较快的检测速度被广泛应用于工业界的目标检测任务中，但是不可置否的是SSD算法的模型参数规模大，模型大小高达100多兆，在GTX1080上的检测速度也有25毫秒之多，这样的效率难以达到实际应用的需求的；MobileNet是近两年提出的轻量化深度神经网络的代表，因为采用了深度可分离卷积，大幅降低了模型参数规模，提高了运算速度，满足了相当一部分轻量化应用的需求。TinyDSOD是将DenseBlock、深度可分离卷积和特征金字塔思想融合起来的深度神经网络，因其速度快正在逐步应用于轻量化的前端检测装置中，但是该网络过度注重速度，模型太小，导致其特征提取能力受限，检测精确度有待进一步提升。

深度神经网络模型以其强大的特征提取能力，在海量的数据支持下，可以实现端到端的目标检测，即检测出目标的同时得到目标的类别。以卡口抓拍图像中的车辆检测为例，现有的方案往往包括以下四步：第一步：对待检目标进行分类，例如分为前拍车身、后拍车身(后续的车辆属性分析多是在前拍车身或后拍车身上进行的)；第二步：收集不同场景不同卡口摄像机的抓拍图像，并对其中的目标按照类别进行标注；第三步：利用标注好的数据训练一个深度神经网络模型；第四步：将已训练好的模型部署到视频分析系统中，实现对卡口抓拍图像中的车辆的检测。

在上述技术方案中，深度神经网络模型的选择是关键。近来，TinyDSOD模型以其良好的性能表现开始工程化应用。TinyDSOD基于主干网络和特征金字塔的思想，将DenseNet中的DenseBlock作为主干网络的基础构件，同时将DenseBlock中的卷积操作替换为深度可分离卷积，在保证网络提取能力的同时，提高检测速度。此外，TinyDSOD中还引入了特征融合的思想，将高层特征向上与邻近的低层特征进行融合，从而提高小目标的检测能力，TinyDSOD的主干网络结构如表1所示。

表1TinyDSOD网络结构

其中的DDB-b如图1所示。

现有技术通常存在以下缺点：

(1)待检类别划分不合理，类内差异大，类间差异小，削弱了深度神经网络的学习能力。如图2和图3所示，公交车的前拍车身和后拍车身差异很小，但它们属于两个类别，大货车的后拍车身和小轿车的后拍车身差异很大，但它们是一个类别。这样的分类方式，导致某些不同类的对象差异太小，或者同一类的对象差异太大，这是目前车辆检测出现误检的重要原因。