[发明专利]证件定位方法、装置、电子设备以及可读存储介质

说明书

本发明实施例提供了一种证件定位方法、装置、电子设备以及可读存储介质，该方法先获取样本图像，其中，样本图像可以通过对目标证件的原始图像进行变换得到，再对样本图像中目标证件的所有顶点坐标进行标注，并基于样本图像构建损失函数，基于该损失函数进行模型参数迭代至收敛，以获得模型定位模块，该模型定位模型采用对目标证件的原始图像进行变换得到的样本图像训练，且对目标证件的所有顶点进行标注，该变换包括旋转、斜切、扭曲、透视、仿射等，使得训练得到的证件定位模型对图像中非规整形状的目标证件能够实现精准的定位，为后续证件中信息的检测、识别、提取等提供了基础，扩大了证件定位的适用性。

技术领域

本发明涉及计算机技术领域，特别是涉及一种证件定位方法、装置、电子设备以及可读存储介质。

背景技术

身份信息的采集和验证在人们的社会活动中被广泛应用，其中，在图像中高效、准确地定位、识别证件信息是身份信息采集验证的关键之一。

目前，相关技术常通过预测边框左上角顶点或中心点的坐标(x，y)以及预测框的宽度w、高度h在图像中定位证件的位置，依据x、y、w、h四个参数可预测得到规整的矩形边框，但是，该方法对图像中非规整矩形的证件无法做到有效检测，定位准确度较低，适用范围小。

发明内容

本发明实施例的目的在于提供一种证件定位方法、装置、电子设备以及可读存储介质，以实现提高证件定位的准确度，对图像中非规整形状的证件有效检测的效果。具体技术方案如下：

在本发明实施的第一方面，首先提供了一种证件定位模型训练方法，该方法可以包括：

获取样本图像，所述样本图像通过对目标证件的原始图像进行变换得到，所述变换包括旋转、斜切、扭曲、透视、仿射中的至少一种；

对所述样本图像中所述目标证件的所有顶点坐标进行标注；

基于所述样本图像构建损失函数，并基于所述损失函数对模型参数进行迭代至收敛，获得证件定位模型。

可选地，所述基于所述样本图像构建损失函数，并基于所述损失函数对模型参数进行迭代至收敛，获得所述证件定位模型，包括：

将所述样本图像分割为第一预设个数的网格；

预测每个所述网格中包含对象，且所述对象为所述目标证件的第一概率值，所述样本图像中包括至少一个对象，所述目标证件为所述对象中的一个；

对每个所述网格预测第二预设个数的边界框，以及所述边界框对应的置信度；

确定所述第一概率值、所述置信度中至少一个对应的损失函数；

根据所述损失函数对所述模型参数进行迭代，至获得符合预设训练条件的所述证件定位模型。

可选地，所述置信度用于表示所述边界框包含所述目标证件的概率，以及基于所述边界框的所述顶点坐标预测所述目标证件的所述顶点坐标的准确度。

可选地，所述对每个所述网格预测第二预设个数的边界框，以及所述边界框对应的置信度，包括：

确定所述边界框中包括所述目标证件的第二概率值；

根据所述边界框的所述顶点坐标与所述目标证件的所述顶点坐标确定交并比；

根据所述第二概率值与所述交并比确定所述边界框对应的置信度。

可选地，所述将所述样本图像分割为第一预设个数的网格，包括：

将所述样本图像缩放至预设尺寸，并将缩放后的所述样本图像分割为第一预设个数的网格。

在本发明实施的第二方面，还提供了一种证件定位方法，该方法可以包括：

获取待测图像；