[发明专利]一种文本识别方法和装置

权利要求书

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

获取第一图像数据，所述第一图像数据包括任意方向的第一文字像素；

通过文本识别模型的文字行检测分支检测所述第一文字像素，获得第一排版方向以及第一旋转角度；

通过所述文本识别模型的文字行识别分支调整所述第一文字像素的第一旋转角度为预设旋转角度；

通过所述文字行识别分支对不同所述第一排版方向对应的所述第一文字像素分别进行文本识别，获得第一识别文本。

2.根据权利要求1所述的方法，其特征在于，所述文字行识别分支包括排版识别子分支，所述通过所述文字行识别分支对不同所述第一排版方向对应的所述第一文字像素分别进行文本识别，获得第一识别文本，包括：

通过所述文字行识别分支确定所述第一排版方向对应的排版识别子分支；

通过所述排版识别子分支对所述第一排版方向对应的所述第一文字像素进行识别，获得第一识别文本。

3.根据权利要求1所述的方法，其特征在于，所述通过所述文本识别模型的文字行识别分支调整所述第一文字像素的第一旋转角度为预设旋转角度，包括：

通过所述文本识别模型的所述文字行识别分支调整所述第一文字像素的第一旋转角度为0。

4.根据权利要求1所述的方法，其特征在于，所述文本识别模型通过如下步骤训练得到：

获取第一预设文本对应的第二图数据，所述第二图像数据包括第二排版方向和第二旋转角度的第二文字像素；

根据所述第二文字像素、所述第二排版方向和所述第二旋转角度对所述文本识别模型中的所述文字行检测分支进行训练；

根据所述第一预设文本、所述第二文字像素、所述第二排版方向和所述第二旋转角度对所述文本识别模型中的所述文字行识别分支进行训练。

5.根据权利要求4所述的方法，其特征在于，所述根据所述第一预设文本、所述第二文字像素、所述第二排版方向和所述第二旋转角度对所述文本识别模型中的所述文字行识别分支进行训练之后，还包括：

获取第二预设文本对应的第三图数据，所述第三图像数据包括第三排版方向和第三旋转角度的第三文字像素；

获取所述文字行检测分支检测所述第三文字像素得到的测试排版方向和测试旋转角度，并根据所述第三排版方向、所述第三旋转角度、所述测试排版方向和所述测试旋转角度调整所述文字行检测分支的参数；

获取所述文字行识别分支根据所述测试排版方向对所述第三文字像素进行识别得到的测试识别文本，并根据所述测试识别文本和所述第二预设文本调整所述文字行识别分支的参数；

根据所述测试识别文本与所述第二预设文本，调整所述文字行检测分支的参数。

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

数据获取模块，用于获取第一图像数据，所述第一图像数据包括任意方向的第一文字像素；

排版检测模块，用于通过文本识别模型的文字行检测分支检测所述第一文字像素，获得第一排版方向以及第一旋转角度；

排版调整模块，用于通过所述文本识别模型的文字行识别分支调整所述第一文字像素的第一旋转角度为预设旋转角度；

文本识别模块，用于通过所述文字行识别分支对不同所述第一排版方向对应的所述第一文字像素分别进行文本识别，获得第一识别文本。

7.根据权利要求6所述的装置，其特征在于，所述文字行识别分支包括至少一个排版识别子分支，所述文本识别模块，包括：

排版对应子模块，用于通过所述文字行识别分支确定所述第一排版方向对应的排版识别子分支；

文本识别子模块，用于通过所述排版识别子分支对所述第一排版方向对应的所述第一文字像素进行识别，获得第一识别文本。

8.根据权利要求6所述的装置，其特征在于，所述排版调整模块，具体用于通过所述文本识别模型的所述文字行识别分支调整所述第一文字像素的第一旋转角度为0。