[发明专利]一种高速公路拥堵识别模型构建方法和装置及识别方法

说明书

本发明公开了一种高速公路拥堵识别模型构建方法和装置及识别方法，步骤1，获取高速公路视频帧图像，得到初始图像集；对初始图像集中的每张图像进行标签标注，获得标签集，所述标签包括交通类别，交通类别包括拥堵、饱和或畅通状态；步骤2，将初始图像集作为输入，标签集作为输出，训练深度卷积神经网络，网络结构在VGG‑16的基础上融入ResNet网络中的经典结构Bottleneck Layer和Squeeze and Excitation(SE)Block，同时引入注意力机制，构建SE‑VGG16的分类网络；本发明能准确进行高速公路交通拥堵状态的识别，可应用于多种交通场景和多种相机视角情况下的交通拥堵识别，具有端到端的特性，实现较为简单且具有较高的识别精度。

技术领域

本发明属于智能交通技术领域，具体涉及一种高速公路拥堵识别模型构建方法和装置及识别方法。

背景技术

交通拥堵检测对于监控交通状况和优化路网性能非常重要。早期交通监控系统中使用了环形线圈检测器来对车辆行驶进行计数，采集交通流的占有率，但是其提供丰富而准确的交通信息的能力非常有限。为了提升准确率，研究人员从GPS跟踪器或智能手机收集的GPS数据，结合路线图来检测交通拥堵。但是这些方法都是对路面具有破坏性的，且需要依赖专门构建和部署的资源，数据较难获取。

随着监控相机安装成本不断降低，路网中大量相机每天都能产生海量监控数据，通过分析这些视频数据提取所需交通流参数，对于交通拥堵状态检测具有重要的实际意义，并且这种方法具有无损性。传统方法通过图像处理的方式，对视频中的车辆进行检测和跟踪，统计交通流参数，这种方法对于遮挡的鲁棒性较低，易导致交通流参数计算错误，影响拥堵状态的识别结果。近年来，卷积神经网络在图像分类和识别方面取得了优异的结果，它们能够自动学习到图像中的特征，并且对于平移、缩放和旋转具有较强的鲁棒性。因此，有一些研究人员将卷积神经网络应用于交通拥堵识别。其中，一类方法是通过网络估计交通流量密度，这种方法易受相机视角变化的影响，另一类方法就是直接通过分类网络计算拥堵程度，但是目前的方法数据只包含特定城市和相机视角，通用性一般。

发明内容

针对现有技术中的缺陷和不足，本发明提供了一种高速公路拥堵识别模型构建方法和装置及识别方法，实现端到端识别，克服现有方法效率低等缺陷。

为达到上述目的，本发明采取如下的技术方案：

一种高速公路拥堵识别模型构建方法，包括以下步骤：

步骤1，获取高速公路视频帧图像，得到初始图像集；

对初始图像集中的每张图像进行标签标注，获得标签集，所述标签包括交通类别，交通类别包括拥堵、饱和或畅通状态；

步骤2，将所述初始图像集作为输入，标签集作为输出，训练深度卷积神经网络；

所述深度卷积神经网络包括依次设置的多个特征提取层和分类层；所述特征提取层包括5个特征提取块，分类层包括3个全连接层；

所述特征提取块包括串联的Bottleneck Layer、SE Block和池化层；

所述Bottleneck Layer包括依次串联的1×1、3×3、1×1大小的3个卷积层；

所述SE Block包括1个卷积层用于对Bottleneck Layer输出的特征图进一步进行特征提取以及1个全局平均池化层；

所述全局平均池化层对应Squeeze、Excitation和Reweight阶段，其中，Squeeze和Excitation阶段用于生成attention机制所需要的权重参数，Reweight阶段将全局平均池化层作用于SE Block的卷积层作用所得特征图，生成具有能够反映不同重要性的特征图；

所述特征提取层的5个特征提取块串联后连接分类层的3个全连接层，用于最后的交通类别分类；