[发明专利]卡证图像倾斜校正方法、装置、计算机设备和存储介质

说明书

本申请涉及一种卡证图像倾斜校正方法、装置、计算机设备和存储介质。本申请能够面对包含复杂背景的卡证图像时，提高卡证校正识别准确率，与传统技术相比，节约处理器计算资源。该方法包括：获取包含卡证的待检测图像；将待检测图像输入训练好的卡证顶点检测模型进行识别，得到卡证的多个卡证顶点坐标；将待检测图像输入训练好的卡证检测模型进行检测，得到卡证对应的拟合矩形框；基于各个卡证顶点坐标与拟合矩形框的位置信息之间的关系，对多个卡证顶点坐标进行分组，以使每组顶点的卡证顶点坐标归属于同一卡证；利用训练好的卡证图像分类网络针对同一卡证进行分类，得到分类后的卡证；若卡证的类别为内容倒立卡证，则将卡证旋转为内容正立卡证。

技术领域

本申请涉及图像识别技术领域，特别是涉及一种卡证图像倾斜校正方法、装置、计算机设备和存储介质。

背景技术

随着数字化信息技术的发展，各行各业纷纷采用证件自动识别并检测证件内容的方式，不需要人们自己手动输入证件内容，大大提高了生活水平以及工作效率。

在证件内容识别过程中，往往需要通过摄像装置先获取证件持有者摆放的卡证，由于证件持有者所处的环境条件(例如光线、桌面背景等)以及摆放方式(手持或者摆放于平面上)不同，导致摄像装置获取的卡证图像常常不规范，例如背景复杂、摆放位置不正等，为了正确读取证件中的文本或图像内容，需要先将卡证图像校正。目前普遍采用基于边缘检测的卡证倾斜校正算法以及基于深度学习的图像分割算法。

然而，这些算法在面对包含复杂背景的卡证图像时，往往需要较大的计算量，且校正准确率不够高。

发明内容

基于此，有必要针对上述技术问题，提供一种卡证图像倾斜校正方法、装置、计算机设备和存储介质。

一种卡证图像倾斜校正方法，所述方法包括：

获取包含卡证的待检测图像；

将所述待检测图像输入训练好的卡证顶点检测模型进行识别，得到所述卡证的多个卡证顶点坐标；

将所述待检测图像输入训练好的卡证检测模型进行检测，得到所述卡证对应的拟合矩形框；

基于各个卡证顶点坐标与所述拟合矩形框的位置信息之间的关系，对所述多个卡证顶点坐标进行分组，以使每组顶点的卡证顶点坐标归属于同一卡证；

利用训练好的卡证图像分类网络针对所述同一卡证进行分类，得到分类后的卡证；

若所述卡证的类别为内容倒立卡证，则将所述卡证旋转为内容正立卡证。

在其中一个实施例中，所述利用训练好的卡证图像分类网络针对所述同一个卡证进行分类，得到分类后的卡证之前，包括：

将所述同一卡证作为当前卡证；

针对所述当前卡证的多个卡证顶点坐标进行大小排序，根据排序结果确定所述当前卡证的各个卡证顶点的位置属性；

根据所述各个卡证顶点的位置属性计算卡证边长，基于所述各个卡证顶点与所述卡证边长的关系，确定所述当前卡证的摆放方向属性。

在其中一个实施例中，所述卡证边长包括上边长、下边长、左边长和右边长；所述基于所述各个卡证顶点与所述卡证边长的关系，确定所述当前卡证的摆放方向属性，包括：

若所述上边长与所述下边长之和大于所述左边长与所述右边长之和，则确定所述当前卡证的摆放方向属性为竖向摆放；

若所述左边长与所述右边长之和大于所述上边长与所述下边长之和，则确定所述当前卡证的摆放方向属性为横向摆放。

在其中一个实施例中，所述方法还包括：

基于所述当前卡证的摆放方向属性，利用投射变换模型，将所述当前卡证变换为与所述摆放方向属性对应的标准矩形图像。