[发明专利]文本图像处理方法、装置、电子设备及可读存储介质

权利要求书

1.一种文本图像处理方法，其特征在于，所述方法包括：

获取初始的待识别文本图像；其中，所述初始的待识别文本图像包括待检测字符图像和可辨识的字符图像，所述待检测字符图像包括不可辨识的字符图像；

根据所述不可辨识的字符图像和所述可辨识的字符图像，检测所述待检测字符图像中字符的位置；

对所述待检测字符图像进行预测得到预测字符图像；

使用所述预测字符图像替换所述待检测字符图像中字符的位置，得到替换的待识别文本图像；

对所述替换的待识别文本图像进行文本识别；

所述对所述待检测字符图像进行预测得到预测字符图像；使用所述预测字符图像替换所述待检测字符图像 中字符的位置，得到替换的待识别文本图像包括：

将所述初始的待识别文本图像中的待检测字符图像打上掩膜；

将所述掩膜后的待识别文本图像输入至图像预测模型，由所述图像预测模型输出所述替换后的待识别文本图像；

所述图像预测模型的训练方法包括：

获取文本图像样本；

随机选取所述文本图像样本中的字符图像，得到随机字符图像；

将所述随机字符图像进行掩膜处理，得到掩膜处理后的文本图像样本；

获取第二训练样本；其中，所述第二训练样本包括所述掩膜处理后的文本图像样本和所述文本图像样本；

根据所述第二训练样本对初始的图像预测模型进行训练，得到所述图像预测模型；其中，所述初始的图像预测模型的输入为所述掩膜处理后的文本图像样本，输出为预测的文本图像，标签为所述文本图像样本。

2.根据权利要求1所述的文本图像处理方法，其特征在于，检测所述待检测字符图像中字符的位置包括：

将所述初始的待识别文本图像输入至目标检测模型，通过所述目标检测模型得到所述待检测字符图像中字符的位置；

其中，所述目标检测模型的输出为两个通道的特征图，其中一个通道表示字符中心点得分图，另一个通道表示字符框的最小外接圆半径。

3.根据权利要求2所述的文本图像处理方法，其特征在于，所述目标检测模型的训练方法包括：

获取第一训练样本；其中，所述第一训练样本包括多个字符图像样本和所述多个字符图像样本的字符位置；

获取初始的目标检测模型；

根据所述第一训练样本对所述初始的目标检测模型进行训练，得到所述目标检测模型；其中，所述初始的目标检测模型的输入为多个字符图像样本，输出为所述多个字符图像样本的预测字符位置，标签为所述多个字符图像样本的字符位置。

4.根据权利要求3所述的文本图像处理方法，其特征在于，

获取第一训练样本包括：获取多个待识别文本图像样本；其中，所述多个待识别文本图像样本包括多个可辨识的字符图像样本和/或多个不可辨识的字符图像样本；根据所述多个待识别文本图像样本获取所述第一训练样本。