[发明专利]图像方向纠正方法、装置及电子设备

说明书

本公开提供了一种图像方向纠正方法、装置及电子设备，涉及人工智能领域。该方法包括：获取待纠正图像，通过图像处理模型对所述待纠正图像中的目标对象进行特征提取，以获取与所述目标对象对应的检测信息；对所述检测信息进行矢量化处理，以获取与所述目标对象对应的坐标信息；基于所述坐标信息确定与所述待纠正图像对应的旋转角度，并根据所述旋转角度对所述待纠正图像的方向进行纠正。本公开能够对存在倾斜的图像进行细粒度旋转角度的纠正，提高了图像方向的纠正精准度；同时能够提高对文档图像中信息的检测效率和识别准确度，进而大大提升了后续信息结构化处理的效率和准确度。

技术领域

本公开涉及计算机技术领域，具体而言，涉及一种图像方向纠正方法、图像方向纠正装置、计算机存储介质及电子设备。

背景技术

随着计算机技术的迅速发展，各个行业的业务逐渐由人工处理模式转换为机器处理模式。并且随着数据量的爆炸式增长，人们对机器处理的效率和精准度提出了越来越高的要求。

以提取文档信息为例，由于拍摄时的角度问题或者其它因素的影响，文档图像存在倾斜，为了保证文档信息的识别效率和精准度，需要对倾斜的文档图像进行纠正。目前对文档图像的纠正主要是根据有限几个固定离散值对图像进行旋转，例如只能旋转90°、180°等等，但是这种方法具有较大局限性，对于存在其它倾斜角度的文档图像，纠正效果较差。

鉴于此，本领域亟需开发一种新的图像方向纠正方法。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

发明内容

本公开的实施例提供了一种图像方向纠正方法、图像方向纠正装置、计算机存储介质及电子设备，进而至少在一定程度上可以对倾斜图像进行细粒度方向纠正，提高图像方向纠正的精准度。

本公开的其他特性和优点将通过下面的详细描述变得显然，或部分地通过本公开的实践而习得。

根据本公开实施例的一个方面，提供了一种图像方向纠正方法，包括：获取待纠正图像，通过图像处理模型对所述待纠正图像中的目标对象进行特征提取，以获取与所述目标对象对应的检测信息；对所述检测信息进行矢量化处理，以获取与所述目标对象对应的坐标信息；基于所述坐标信息确定与所述待纠正图像对应的旋转角度，并根据所述旋转角度对所述待纠正图像的方向进行纠正。

根据本公开实施例的一个方面，提供了一种图像方向纠正装置，包括：检测信息获取模块，用于获取待纠正图像，通过图像处理模型对所述待纠正图像中的目标对象进行特征提取，以获取与所述目标对象对应的检测信息；坐标信息获取模块，用于对所述检测信息进行矢量化处理，以获取与所述目标对象对应的坐标信息；图像方向纠正模块，用于基于所述坐标信息确定与所述待纠正图像对应的旋转角度，并根据所述旋转角度对所述待纠正图像的方向进行纠正。

在本公开的一些实施例中，所述图像处理模型包括特征提取子模型、特征融合子模型和后处理子模型；基于前述方案，所述检测信息获取模块包括：目标特征信息获取单元，用于通过所述特征提取子模型对所述目标对象进行多层卷积，以获取多级目标特征信息；目标融合特征信息获取单元，用于通过所述特征融合子模型根据各级所述目标特征信息进行特征融合，以获取目标融合特征信息；检测信息获取单元，用于通过所述后处理子模型对所述目标融合特征信息进行卷积处理，以获取与所述目标对象对应的检测信息。