[发明专利]一种多角度车型识别方法及系统

说明书

本发明公开了一种多角度车型识别方法及系统，该方法包括以下步骤：收集多角度车辆图片，构建多角度车辆数据库，进行预处理获得多角度车辆训练图片，未经预处理的图片作为测试样本图片；构建卷积神经网络，在所述卷积神经网络中增加注意力机制，使用深度可分离卷积代替传统卷积；对所述多角度车辆训练图片进行卷积神经网络训练，获得多角度车辆识别模型；通过所述车辆识别模型对所述多角度车辆数据库内的所述测试样本图片进行分类识别。

技术领域

本发明涉及模式识别与图像处理技术领域，具体的说，是一种多角度车型识别方法及系统。

背景技术

多角度车型识别是智能交通系统的重要组成部分之一，具有更加广泛的应用，包括智能停车系统、车辆流量的统计和车辆的检测。随着交通监控摄像机的大量使用，车型识别在计算机视觉领域得到了极大的关注，在大型卷积神经网络提取高维度特征过程中，多角度计算复杂度较高且准确率相对较低。

发明内容

有鉴于此，本发明解决上述技术问题的技术方案如下：一种多角度车型识别方法及系统。本发明为了解决现存问题，提供以下技术方案：

一种多角度车型识别方法，包括以下步骤：

收集多角度车辆图片，构建多角度车辆数据库，进行预处理获得多角度车辆训练图片，未经预处理的图片作为测试样本图片；

构建卷积神经网络，在所述卷积神经网络中增加注意力机制，使用深度可分离卷积代替传统卷积；

对所述多角度车辆训练图片进行卷积神经网络训练，获得多角度车辆识别模型；

通过所述车辆识别模型对所述多角度车辆数据库内的所述测试样本图片进行分类识别。

进一步，所述进行预处理获取多角度车辆训练图片，具体包括以下步骤：

将所述多角度车辆图片剪裁成统一大小，以及对所述剪裁成统一大小的多角度车辆图片进行旋转和镜像，得到数据增强后的样本图片；

将所述数据增强后的样本图片进行灰度化处理，得到多角度车辆训练图片。

进一步，所述构建卷积神经网络，在所述卷积神经网络中增加注意力机制，使用深度可分离卷积代替传统卷积，具体包括以下步骤：

构建卷积神经网络，所述卷积神经网络包括数据输入层、Resnet18网络卷积层、特征层、全连接特征提取层、Drop_out层、全连接分类层和分类损失层；

在所述卷积神经网络中增加注意力机制结构，依次连接Conv1层、Conv2层、全局平均池化层、Fc1层、ReLU层、Fc2层、Sigmoid层，以及与所述Conv2层相连接的Scale层；

在所述卷积神经网络中，将传统卷积核为3*3的卷积层替换成卷积核为3*3的深度卷积层和卷积核为1*1的逐点卷积层，所述深度卷积和逐点卷积合称为深度可分离卷积，在1*1逐点卷积运算后，不使用ReLU激活函数。

进一步，所述注意力机制具体包括：

特征压缩阶段：使用所述全局平均池化层，将所述卷积层的每个通道的二维特征变成一个实数，对所述二维特征进行压缩；

特征激励阶段：使用一个全连接层，将输入的所述二维特征的通道数降低，减少通道的个数，降低计算量；连接ReLU激活函数，保持输出维度不变，得到非线性特征；再通过一个全连接层变回原来的维度，经过Sigmoid函数将特征权重归一化到0-1之间；

特征重标定阶段：使用所述缩放层对每个所述二维特征的通道的权重进行重新标定，将所述二维特征激励操作后输出的权重加权到每个所述通道的二维特征上，以此完成对原始二维特征的重新标定。

进一步，所述深度可分离卷积用于计算复杂度，所述深度可分离卷积计算复杂度具体包括：