[发明专利]倾斜对象处理方法及装置

说明书

本公开是关于一种倾斜对象处理方法及装置。该方法包括：将待检测图像输入倾斜对象检测网络模型中进行处理，确定待检测图像中的倾斜对象的类别信息、位置信息以及倾斜角度信息；根据待检测的图像中倾斜对象的类别信息、位置信息以及倾斜角度信息，校正倾斜对象，其中，倾斜对象检测网络模型包括特征提取网络、区域生成网络RPN、区域特征提取网络以及目标检测网络。根据本公开实施例，通过倾斜对象检测网络模型确定待检测图像中倾斜对象的类别信息、位置信息以及倾斜角度信息，并根据倾斜对象的类别信息、位置信息以及倾斜角度信息，对倾斜对象进行校正，从而实现倾斜对象的准确校正。

技术领域

本公开涉及计算机技术领域，尤其涉及一种倾斜对象处理方法及装置。

背景技术

随着信息技术的快速发展，数字化文档、图片所带来的易检索、存储以及网络传输的优势也越发明显。例如，人们可以将各种纸质图片、文档都通过数字化扫描方式进行存储，以满足存储、网络传输等需求。企业也利用数字化文档、图片来保存相关信息，例如，银行存储包括用户电子签名的文档等。然而，数字化文档、图片中易出现倾斜的现象，给日常使用带来不便。

发明内容

为克服相关技术中存在的问题，本公开提供一种倾斜对象处理方法及装置。

根据本公开实施例的第一方面，提供一种倾斜对象处理方法，所述方法包括：

将待检测图像输入倾斜对象检测网络模型中进行处理，确定所述待检测图像中的倾斜对象的类别信息、位置信息以及倾斜角度信息；

根据待检测的图像中倾斜对象的类别信息、位置信息以及倾斜角度信息，校正所述倾斜对象，

其中，所述倾斜对象检测网络模型包括特征提取网络、区域生成网络RPN(RegionProposal Network)、区域特征提取网络以及目标检测网络。

对于以上方法，在一种可能的实现方式中，将待检测图像输入倾斜对象检测网络模型中进行处理，确定所述待检测图像中的倾斜对象的类别信息、位置信息以及倾斜角度信息，包括：

将所述待检测图像输入所述特征提取网络中进行处理，确定所述待检测图像的第一特征信息；

将所述第一特征信息输入到区域生成网络RPN中进行处理，确定所述待检测图像的多个候选区域；

将所述第一特征信息以及所述候选区域输入到所述区域特征提取网络中进行处理，确定所述待检测图像的多个候选区域的第二特征信息；

将所述第二特征信息输入到所述目标检测网络中进行处理，确定所述待检测图像中倾斜对象的类别信息、位置信息以及倾斜角度信息。

对于以上方法，在一种可能的实现方式中，所述目标检测网络包括用于检测倾斜对象的类别信息和位置信息的第一目标检测网络以及用于检测倾斜对象的倾斜角度信息的第二目标检测网络，

将所述第二特征信息输入到所述目标检测网络中进行处理，确定所述待检测图像中倾斜对象的类别信息、位置信息以及倾斜角度信息，包括：

将所述第二特征信息输入到所述第一目标检测网络中进行处理，确定所述待检测图像中倾斜对象的类别信息以及位置信息；

将所述位置信息以及所述第二特征信息输入到所述第二目标检测网络中进行处理，确定所述待检测图像中倾斜对象的倾斜角度信息。

对于以上方法，在一种可能的实现方式中，所述方法还包括：

获取包括倾斜对象的样本图像；

通过多维向量标记所述样本图像中倾斜对象的位置信息以及倾斜角度信息；

将标记的样本图像输入到倾斜对象检测网络模型中，对所述倾斜对象检测网络模型进行训练，确定所述倾斜对象检测网络模型中的参数权重。