[发明专利]一种基于二维高斯边界框重叠度度量的车牌检测方法

说明书

本发明属于车牌检测技术领域，涉及一种基于二维高斯边界框重叠度度量的车牌检测方法，通过将朝向边界框的转换为二维高斯分布的表示，并使用KL散度作为度量标准来计算预测边界框与真实边界框之间的分布距离，以此作为边界框的回归损失，并且该方法是可微的，解决平滑L1度量损失与评估方式不一致的问题，可以提高基于朝向框的车牌检测网络训练效率和泛化能力，并且仅用于网络训练阶段，并不增加推理过程的计算量，不仅可以用来进行倾斜形变车牌检测，还可以用于场景文本检测、超市商品检测等多种场景的目标检测任务。

技术领域

本发明属于车牌检测技术领域，涉及一种基于二维高斯边界框重叠度度量的车牌检测方法。

背景技术

车辆使用的车牌号是车辆唯一的身份标志，也是车辆上路行驶的重要凭证，车牌的检测识别是辨识车辆身份的主要手段。因此车牌检测是智慧交通数据统计分析的重要任务，也是城市车辆管理的前置任务，其应用场景十分广泛，如停车场、小区准入验证和道路交通监控等，然而实际应用场景中车牌收到光照、视角变化等移速的影响，使得检测结果不佳。

现有的车牌检测方法主要有两种，一是基于传统方法的车牌检测方法，通过使用梯度和颜色空间进行车牌检测，这类方法拥有较快的检测速度，但是对复杂的背景和场景中的车牌特征建模差，对遮挡、模糊、形变的车牌定位不准确；二是基于深度学习的车牌检测方法，这类方法通过学习大量的训练数据，对不同的场景具有很强的适应能力，为了更好的描述倾斜车牌的位置，很多深度学习方法使用朝向框的方法定义车牌的边界框，在训练时通常使用平滑L1损失来度量误差，但是这种损失评价方式与评估方式并不一致，即预测框与真实框的重叠程度与损失的大小不是线性的关系；在通用目标检测（水平框）中通常使用重叠度损失（IoU Loss）来解决损失和评估的不一致性，然而朝向框使用重叠度损失会带来损失函数不可导的问题，影响了训练过程。

由此可见，在基于深度学习的车牌朝向框训练过程中，现有的方法存在的损失和评估方式不一致的问题，严重影响了训练的效率和精度，因此急需有效的方法用于提升车牌检测精度。

发明内容

为了克服上述现有技术的不足，本发明设计提供一种基于二维高斯边界框重叠度度量的车牌检测方法，用来解决畸形形变车牌训练过程中损失和评估方式不一致的问题，可用于任意场景的形变车牌训练任务，能够高效的车牌检测的精度和训练效率。

为实现上述目的，本发明首先构架数据集并设计简洁的主干网络，避免任何繁琐的后处理步骤，用于预测的旋转车牌的朝向框，再将基于五点定位法的预测朝向框近似为一个二维高斯分布，通过计算预测朝向框与真实朝向框之间的二维高斯分布之间的距离度量两个框的重叠度，进而实现对朝向框质量的度量，具体过程为：

（1）数据集构建：收集小区出入口、侧方位停车场等场景中含有常规以及倾斜、畸变车牌的图像，构建含有车牌的图像数据集，并标注车牌的位置，再将数据集划分为训练集、验证集和测试集，其中训练集用于训练，验证集用于训练过程中的模型验证，测试集用于最终的精度测试；

（2）主干网络特征提取：先对数据集中的图片尺寸和数值范围进行初始化处理，再将处理后的图像输入到主干网络中进行卷积特征提取；

（3）多尺度特征层次结构：将特征提取后的图像通过基于特征融合的多尺度特征层次结构得到的多尺度特征层次特征图集合；

（4）车牌位置定位：根据步骤（3）得到的多尺度特征层次特征图集合，在不同的多尺度卷积层后连接两层卷积核为3*3的卷积层以及一层平均池化层，在特征图的每一个特征点设置一个锚框，分别使用两个结构相同的但不共享参数的全连接层子网络学习分类和位置偏移信息，实现目标框的类别分类和位置回归；其中位置偏移信息是车牌的目标框与特征点对应的锚框的偏差，类别分类是z判断是否是车牌，位置为车牌的朝向框，其中表示车牌的中心点，表示车牌朝向框的长和宽，表示与水平方向的夹角；

（5）朝向框转换：将基于五点表示法的朝向框转换成基于二维高斯分布表示的朝向框，获得预测的朝向框和真实的朝向框的二维高斯分布形式；