基于深度学习的车辆分类与计数方法与流程

本发明涉及车辆分类与计数方法，特别是涉及基于深度学习的车辆分类与计数方法。

背景技术：

近年来，随着交通车辆数量高速增长，交通监管面临巨大挑战。随着智能城市智能交通系统(its)的推广，其应用的核心技术得到了迅速的发展。车辆目标检测，作为构建交通状况视频监控的一项关键技术，一直是其重要的组成部分。车流量检测的大致的可以归于三类：第一类是在地下埋感应线圈，当车辆经过感应线圈区域时，由感应线圈装置受到压力发出脉冲信号，从而判断是否有车辆。常用的方法有环形线圈检测法、地磁检测法、电磁检测法等。这类方法较在安装和维修时需要破坏地面，且费用较高，同时也给车辆的出行造成了不便。第二类是悬挂式检测方法，当车辆驶入检测范围时，检测器发出的微波、超声波或红外线，以此来判断是否有车辆。这类检测方法易于安装，对路面不会造成破坏，但当两辆车并排进入检测范围时，只能计数一辆车，检测精度受到了影响，检测范围较小。这类检测方法易于安装，对路面不会造成破坏，但当两辆车并排进入检测范围时，只能计数一辆车，检测精度受到了影响，检测范围较小。第三类为通过摄像机拍摄交通视频，利用计算机视觉技术以及图像处理技术对视频进行处理。相对于前两类传统方法，视频检测方法具有明显的优势：安装和维护装置方便，能对整个路面的路况进行拍摄，检测到的范围比较广，获取到的交通信息比较全面。基于视觉的车辆检测是目前最流行的方法。传统的基于视觉的车辆检测方法通常是手动提取特征，费时费力，泛化能力差，易受环境变化和遮挡的影响。随着深度学习理论和实践的迅速发展，基于机器学习的目标检测与分类进入了一个新的阶段。不同于传统的特征提取算法，卷积神经网络具有一定的不变性的几何变换、变形和光照，可以克服改变车辆外观的困难，可以解决遮挡问题，能够自适应的训练数据驱动下构建的特征描述，具有更大的灵活性和综合能力。

虽然基于深度学习的目标检测具有一些优点，但是其速度通常不是很快。近年来，如ssd，yolo2和其他快速目标检测网络，在保持高速度的同时实现高精度，在泰坦x下的速度可达每秒几十帧，但实际上，我们很少使用高端显卡，一方面成本太高，另一方面空间太大。在主流计算机中实现其实时性是主要问题。

传统的基于视觉的车辆检测方法通常是手动提取特征，费时费力，泛化能力差，易受环境变化和遮挡的影响。不同于传统的特征提取算法，卷积神经网络具有一定的不变性的几何变换、变形和光照，可以克服改变车辆外观的困难，可以解决遮挡问题，能够自适应的训练数据驱动下构建的特征描述，具有更大的灵活性和综合能力。

现有技术

传统的车辆计数方法使用的是背景相减法,将视频中的每一帧和背景进行对比，并分离出和背景有差异的对象。通常会设置一个虚拟线圈来对车辆进行计数，根据虚拟线圈内像素的变化来对车辆计数。然而单纯的背景相减法，虽然速度快，但在精度上有所欠缺。考虑到实时性，目前大多数的传统技术都只实现了对车辆的计数。然而随着智能城市智能交通系统(its)的发展，车辆分类的信息也成为了不可欠缺的一部分。

技术实现要素：

基于此，有必要针对上述技术问题，提供一种基于深度学习的车辆分类与计数方法，传统的虚拟检测器与singleshotmultiboxdetector结合，使其在能达到实时检测的情况下，达到了高精度。

一种基于深度学习的车辆分类与计数方法，包括：

用singleshotmultiboxdetector训练一个车辆检测模型；

设置一个虚拟线圈，只记录该区域内的用singleshotmultiboxdetector训练的训练一个车辆检测模型检测出来的车辆中心点；

看车辆是否经过虚拟线圈，只有当车辆经过虚拟线圈检测器时，才进行下一步；

如果是第一帧，记录下在虚拟检测器里用singleshotmultiboxdetector检测出来的所有的车辆的中心点；

如果不是第一帧，则和上一帧标记的中心点依次进行比对，如果距离小于一定阈值，则更新该点，否则新建一个点并记录；

当一个中心点连续消失超过n帧时并且在虚拟线圈内存在超过n帧，消去该点，对应类别计数+1，总计数+1；

若中心点连续消失超过n帧但在虚拟线圈内存在不超过n帧，则认为是误识别的点，消去该点不计数。

上述基于深度学习的车辆分类与计数方法，实现了基于深度学习的实时车辆分类与计数；在实现车辆计数的同时，可对车辆进行分类。并且达到实时检测速度。

一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现任一项所述方法的步骤。

一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现任一项所述方法的步骤。

一种处理器，所述处理器用于运行程序，其中，所述程序运行时执行任一项所述的方法。

附图说明

图1为本申请实施例提供的一种基于深度学习的车辆分类与计数方法的流程图。

具体实施方式

为了使本发明的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本发明进行进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本发明，并不用于限定本发明。

一般基于视觉的车辆检测方法都使用了虚拟线圈检测器，即在图像中框出一块区域，当车辆经过该区域时，才开始计数。近年来，使用了多个虚拟检测器来提高精度。类似地，我们也使用虚拟检测器来统计车辆，得益于深度学习的高效性，本文仅使用单个虚拟检测器。传统方法是根据像素点在虚拟线圈检测器内的状况来识别车辆，和那些方法不同，我们使用车辆的中心点在虚拟检测器内的状况来识别车辆。

参阅图1，本发明的流程如下：

用singleshotmultiboxdetector训练一个车辆检测模型

设计一个系统，流程如下：

1)首先看车辆是否经过虚拟线圈，只有当车辆经过虚拟线圈检测器时，才进行下一步；

2)如果是第一帧，记录下在虚拟检测器里用singleshotmultiboxdetector检测出来的所有的车辆的中心点。

3)如果不是第一帧，则和上一帧标记的中心点依次进行比对，如果距离小于一定阈值，则更新该点，否则新建一个点并记录。

4)当一个中心点连续消失超过n帧时并且在虚拟线圈内存在超过n帧，消去该点，对应类别计数+1，总计数+1。若中心点连续消失超过n帧但在虚拟线圈内存在不超过n帧，则认为是误识别的点，消去该点不计数。

以上所述实施例的各技术特征可以进行任意的组合，为使描述简洁，未对上述实施例中的各个技术特征所有可能的组合都进行描述，然而，只要这些技术特征的组合不存在矛盾，都应当认为是本说明书记载的范围。

以上所述实施例仅表达了本发明的几种实施方式，其描述较为具体和详细，但并不能因此而理解为对发明专利范围的限制。应当指出的是，对于本领域的普通技术人员来说，在不脱离本发明构思的前提下，还可以做出若干变形和改进，这些都属于本发明的保护范围。因此，本发明专利的保护范围应以所附权利要求为准。