[发明专利]文本分离方法、装置、电子设备和存储介质

权利要求书

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

确定待分离的重叠文本图像；

将所述重叠文本图像输入至文本分离模型，得到所述文本分离模型输出的文本分离图像；

所述文本分离模型基于样本重叠文本图像，以及所述样本重叠文本图像的样本文本分离图像，联合判别模型进行对抗训练得到，所述判别模型用于区分预测文本分离图像与所述样本文本分离图像，所述预测文本分离图像是所述文本分离模型基于所述样本重叠文本图像确定的。

2.根据权利要求1所述的文本分离方法，其特征在于，所述将所述重叠文本图像输入至文本分离模型，得到所述文本分离模型输出的文本分离图像，包括：

将所述重叠文本图像输入至所述文本分离模型中的特征卷积网络，得到所述特征卷积网络输出的所述重叠文本图像的图像特征；

将所述图像特征输入至所述文本分离模型中的固定文本生成网络，得到所述固定文本生成网络输出的固定文本图像；

将所述图像特征输入至所述文本分离模型中的印刷文本生成网络，得到所述印刷文本生成网络输出的印刷文本图像，所述固定文本图像和所述印刷文本图像构成所述文本分离图像。

3.根据权利要求2所述的文本分离方法，其特征在于，所述将所述图像特征输入至所述文本分离模型中的印刷文本生成网络，得到所述印刷文本生成网络输出的印刷文本图像，包括：

将所述图像特征分别输入至所述印刷文本生成网络下的印刷文本生成分支和印刷文本二值图生成分支，得到所述印刷文本生成分支输出的初始印刷文本图像，以及所述印刷文本二值图生成分支输出的印刷文本二值图；

基于所述印刷文本二值图，以及所述初始印刷文本图像，确定所述印刷文本图像。

4.根据权利要求3所述的文本分离方法，其特征在于，所述判别模型包括固定文本判别分支，印刷文本判别分支以及印刷文本二值图判别分支；

所述固定文本判别分支用于判别固定文本图像的真伪；

所述印刷文本判别分支用于判别初始印刷文本图像的真伪；

所述印刷文本二值图判别分支用于判别印刷文本二值图的真伪。

5.根据权利要求1至3中任一项所述的文本分离方法，其特征在于，所述文本分离模型和所述判别模型基于如下步骤确定：

构建初始文本分离模型和初始判别模型；

将所述样本重叠文本图像输入至所述初始文本分离模型，得到所述初始文本分离模型输出的预测文本分离图像；

将所述预测文本分离图像，以及所述样本文本分离图像分别输入至所述初始判别模型，得到所述初始判别模型输出的所述预测文本分离图像的判别结果，以及所述样本文本分离图像的判别结果；

基于所述预测文本分离图像，所述样本文本分离图像，所述预测文本分离图像的判别结果以及所述样本文本分离图像的判别结果，对所述初始文本分离模型和所述初始判别模型进行参数更新，得到所述文本分离模型和所述判别模型。

6.根据权利要求5所述的文本分离方法，其特征在于，所述基于所述预测文本分离图像，所述样本文本分离图像，所述预测文本分离图像的判别结果以及所述样本文本分离图像的判别结果，对所述初始文本分离模型和所述初始判别模型进行参数更新，得到所述文本分离模型和所述判别模型，包括：

基于所述预测文本分离图像，以及所述样本文本分离图像，确定所述初始文本分离模型的生成损失；

基于所述预测文本分离图像的判别结果，以及所述样本文本分离图像的判别结果，确定所述初始判别模型的判别损失；

基于所述生成损失和所述判别损失，对所述初始文本分离模型和所述初始判别模型进行参数更新，得到所述文本分离模型和所述判别模型。