[发明专利]文本识别方法、装置、电子设备及计算机可读存储介质

权利要求书

1.一种文本识别方法，其特征在于，包括：

获取包含文本信息的待识别图片；

通过预先训练好的文本检测模型对所述待识别图片进行识别，确定所述待识别图片中包含文本的至少一个文本框，及各所述文本框对应的倾斜方向；

根据所述倾斜方向对各所述文本框的文本方向进行校正，得到所述文本方向校正后的校正文本框；

识别所述校正文本框中的文本信息。

2.根据权利要求1所述的方法，其特征在于，在所述获取包含文本信息的待识别图片之前，还包括：

确定预先训练的所述文本检测模型；

所述确定预先训练的所述文本检测模型，包括

获取样本图片；所述样本图片中包含有预先标注的至少一个初始文本框，及各所述初始文本框在所述样本图片中所处的初始位置信息、各所述初始文本框的初始倾斜方向；

将所述样本图片依次输入至初始文本检测模型对所述初始文本检测模型进行训练，确定所述样本图片对应的至少一个预测文本框，及各所述预测文本框在所述样本图片中所处的预测位置信息、各所述预测文本框的预测倾斜方向；

根据各所述初始位置信息、各所述预测位置信息、各所述初始倾斜方向和各所述预测倾斜方向，计算得到所述初始文本检测模型的损失值；

在所述损失值处于预设范围内的情况下，将训练后的初始文本检测模型作为所述文本检测模型。

3.根据权利要求2所述的方法，其特征在于，所述根据各所述初始位置信息、各所述预测位置信息、各所述初始倾斜方向和各所述预测倾斜方向，计算得到所述初始文本检测模型的损失值，包括：

根据各所述初始位置信息和各所述预测位置信息，计算得到位置损失值；

根据各所述初始倾斜方向和各所述预测倾斜方向，计算得到倾斜损失值；

根据所述位置损失值、位置权重、所述倾斜损失值和倾斜权重，计算得到所述初始文本检测模型的损失值。

4.根据权利要求3所述的方法，其特征在于，所述文本检测模型包括：分类结果获取层和倾斜方向获取层，所述通过预先训练的文本检测模型对所述待识别图片进行识别，确定所述待识别图片中包含文本的至少一个文本框，及各所述文本框对应的倾斜方向，包括：

调用所述分类结果获取层对所述待识别图片进行处理，获取所述待识别图片上的像素分类结果和连通分类结果；

根据所述像素分类结果和所述连通分类结果，确定所述待识别图片中的至少一个文本框；

调用所述倾斜方向获取层对所述待识别图片进行处理，确定所述至少一个文本框中文本的待确认倾斜方向，及所述待确认倾斜方向对应的倾斜阈值；

根据所述倾斜阈值中的最大倾斜阈值，确定所述至少一个文本框对应的倾斜方向。

5.根据权利要求1所述的方法，其特征在于，所述根据所述倾斜方向对各所述文本框进行校正处理，得到校正文本框，包括：

在根据所述倾斜方向确定所述文本框的倾斜角度位于90°至180°之间时，将所述文本框逆时针旋转90°，得到所述校正文本框；

在根据所述倾斜方向确定所述文本框的倾斜角度位于180°至270°之间时，将所述文本框逆时针旋转180°，得到所述校正文本框；

在根据所述倾斜方向确定所述文本框的倾斜角度位于270°至360°之间时，将所述文本框逆时针旋转270°，得到所述校正文本框。

6.根据权利要求1所述的方法，其特征在于，所述识别所述校正文本框中的文本信息，包括：

将所述校正文本框输入至文本识别模型，通过所述文本识别模型确定所述校正文本框中包含的文本信息。

7.一种文本识别装置，其特征在于，包括：

待识别图片获取模块，用于获取包含文本信息的待识别图片；

待识别图片识别模块，用于通过预先训练好的文本检测模型对所述待识别图片进行识别处理，确定所述待识别图片中包含文本的至少一个文本框，及各所述文本框对应的倾斜方向；

校正文本框获取模块，用于根据所述倾斜方向对各所述文本框的文本方向进行校正处理，得到所述文本方向校正后的校正文本框；

文本信息确定模块，用于识别所述校正文本框中的文本信息。