[发明专利]图像处理方法及相关设备

说明书

本公开实施例提供了一种图像处理方法及装置、电子设备和存储介质。该方法包括：获取待处理图像；检测所述待处理图像中的目标轮廓线；根据所述目标轮廓线旋转所述待处理图像，使所述待处理图像处于预设方向；对旋转后的所述待处理图像进行分类处理，获得旋转后的所述待处理图像的目标方向类别信息。本公开实施例提供的方法可以减少图片分类的类别数量，提升分类效果。

技术领域

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

背景技术

在对名片的OCR(Optical Character Recognition，光学字符识别)识别场景中，由于拍摄的图片中的名片存在旋转的情况，可能会导致无法识别出图片中的文字，因此，需要在图片预处理中对存在旋转情况的图片进行调整。

在图片预处理中可以根据图片旋转角度对旋转的图片进行分类，图片的旋转角度可以从0到360°，将图片分为360类工作量大，且分类效果差。

因此，需要一种新的图像处理方法及装置、电子设备和计算机可读存储介质。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解。

发明内容

本公开实施例提供一种图像处理方法及装置、电子设备和计算机可读存储介质，该方法可以减少图片分类的类别数量，提升分类效果。

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

本公开实施例提供一种图像处理方法，该方法包括：获取待处理图像；检测待处理图像中的目标轮廓线；根据目标轮廓线旋转待处理图像，使待处理图像处于预设方向；对旋转后的待处理图像进行分类处理，获得旋转后的待处理图像的预设方向的目标方向类别信息，其中，目标方向类别信息是根据待处理图像的内容方向确定的。

本公开实施例提供一种图像处理装置，该装置包括：图像获取模块，用于获取待处理图像；轮廓线检测模块，用于检测待处理图像中的目标轮廓线；图像旋转模块，用于根据目标轮廓线旋转待处理图像，使待处理图像处于预设方向；图像处理模块，用于对旋转后的待处理图像进行分类处理，获得旋转后的待处理图像的目标方向类别信息，其中，目标方向类别信息是根据待处理图像的内容方向确定的。

在本公开的一些示例性实施例中，图像旋转模块包括：夹角获取单元，用于分别获取第一夹角及第二夹角，其中，第一夹角为目标轮廓线与第一方向的夹角，第二夹角为目标轮廓线与第二方向的夹角，第一夹角与第二夹角均为锐角；方向确定单元，用于根据第一夹角及第二夹角，确定第一方向或者第二方向为目标方向；图像旋转单元，用于旋转待处理图像至预设方向，以使目标轮廓线与目标方向平行。

在本公开的一些示例性实施例中，图像处理装置还包括：训练集获取单元，用于获取训练集，训练集包括训练图像及其方向类别标签，训练图像中包含第一语言名片、第二语言名片及双语名片，训练图像处于预设方向；图像处理单元，用于通过神经网络模型对训练图像进行处理，获得训练图像的预测方向类别信息；模型训练单元，用于根据方向类别标签及预测方向类别信息，训练神经网络模型，确定神经网络模型的目标网络参数，将包括目标网络参数的神经网络模型确定为图片分类模型；其中，图片分类模型用于对旋转后的待处理图像进行分类处理，获得旋转后的待处理图像的目标方向类别信息。

在本公开的一些示例性实施例中，图像处理装置还包括：自然图像获取单元，用于获取目标数据库中的自然图像及标签信息；自然图像处理单元，用于通过初始神经网络模型对所述自然图像进行处理，获得初始预测信息；模型获得单元，用于根据所述标签信息及所述初始预测信息训练所述初始神经网络模型，以获得所述神经网络模型。