[发明专利]一种车牌类型识别模型构建及车牌类型识别方法

说明书

本发明公开了一种车牌类型识别模型构建及车牌类型识别方法，包括：获取多个不同种类的车牌图像作为训练样本；对训练样本进行均匀采样，将采样后得到的车牌图像输入到第一神经网络模型进行训练得到第一车牌类型识别模型；确定所述训练样本中每一种车牌图像的数量；对数量满足第一预设数量条件的车牌图像进行过采样处理；对数量满足第二预设数量条件的车牌图像进行欠采样处理；利用过采样处理后得到的车牌图像和欠采样处理后得到的车牌图像构成的训练样本训练第二神经网络模型得到第二车牌类型识别模型；将第一车牌类型识别模型和第二车牌类型识别模型进行融合，得到用于进行车牌类型识别的骨干网络。

技术领域

本发明涉及车辆识别技术领域，具体涉及一种车牌类型识别模型构建及车牌类型识别方法。

背景技术

在智能交通应用场景中，一般通过对采集到的车辆图像中车牌号码的识别来确定场景中车辆的身份，但当采集到的车辆图像的分辨率较低或车牌号码局部被遮挡，可能导致车牌号码识别错误，甚至无法识别出车牌号码，当遇到突发交通情况时，影响对场景中车辆身份的确认。由于机动车车牌分为多个类型，如可以包括普通蓝牌、单层黄牌、教练车牌、警用车牌、新能源白绿牌、新能源黄绿牌、挂车牌、双层黄牌、临行车牌等类型，通过识别车牌类型属性可以进一步缩小车辆查询范围。

在进行训练样本采样以用于训练机器学习模型的过程中，由于目前道路上行驶的车辆中普通蓝牌和新能源车牌占据较大比例，而警用车、挂车牌、双层黄牌、临行车牌等类型的车牌较少，因此造成采集到的训练样本存在不均衡的问题，继而导致训练得到的机器学习模型对车牌类型的识别结果的准确性。

发明内容

因此，本发明要解决的技术问题在于克服现有对车牌类型的识别结果准确性差的缺陷，从而提供一种车牌类型识别模型构建及车牌类型识别方法。

根据第一方面，本发明实施例公开了一种车牌类型识别模型构建方法，包括：获取多个不同种类的车牌图像作为训练样本；对所述训练样本进行均匀采样，将采样后得到的车牌图像输入到第一神经网络模型进行训练，得到第一车牌类型识别模型；确定所述训练样本中每一种车牌图像的数量；对数量满足第一预设数量条件的车牌图像进行过采样处理；对数量满足第二预设数量条件的车牌图像进行欠采样处理，其中所述第二预设数量条件对应的数量值大于所述第一预设数量条件对应的数量值；利用过采样处理后得到的车牌图像和欠采样处理后得到的车牌图像构成的训练样本训练第二神经网络模型，得到第二车牌类型识别模型；将所述第一车牌类型识别模型和所述第二车牌类型识别模型进行融合，得到用于进行车牌类型识别的骨干网络。

可选地，所述获取多个不同种类的车牌图像作为训练样本，包括：获取每一种车辆在多个场景下、多个时间段内不同角度的车辆图像，每一个车辆图像包含的车牌图像满足预设清晰度和完整度要求；利用预设关键点定位方法定位得到每一个车辆图像中的车牌图像；根据定位得到的车牌图像的关键点计算车牌的水平倾斜角度与垂直倾斜角度；对定位得到的每一个车辆图像中的车牌图像进行抠图处理，将得到的车牌图像按照所述水平倾斜角度与垂直倾斜角度进行旋转处理；对旋转处理后得到的车牌图像进行标签化处理得到所述训练样本。

可选地，将所述第一车牌类型识别模型和所述第二车牌类型识别模型进行融合，得到用于进行车牌类型识别的骨干网络，包括：获取所述第一车牌类型识别模型中每一个目标层的第一权重以及所述第二车牌类型识别模型中与所述目标层相对应的层的第二权重；利用预设调节参数对所述第一权重和所述第二权重进行预处理，得到共享权重；利用所述共享权重替换所述第一车牌类型识别模型中所述目标层的第一权重，利用所述共享权重替换所述第二车牌类型识别模型中与所述目标层相对应的层的第二权重；将权重替换处理后的第一车牌识别模型和权重替换处理后的第二车牌类型识别模型进行融合处理，得到用于进行车牌类型识别的骨干网络。

可选地，所述利用预设调节参数对所述第一权重和所述第二权重进行预处理，得到共享权重，包括：按照下式进行预处理：

ω＝λω₁+(1-λ)ω₂