[发明专利]文字识别网络训练方法、装置、计算机设备及存储介质

权利要求书

1.一种文字识别网络训练方法，其特征在于，包括：

获取带有文字的样本图像集，其中，所述样本图像集中包括对应有正确文本标注信息和易混淆文本标注信息的第一样本图像；

基于待训练的第一文字识别网络，确定所述第一样本图像对应的预测文本内容；

基于所述正确文本标注信息和所述预测文本内容，确定以所述正确文本标注信息为监督信息时的第一损失值；以及基于所述易混淆文本标注信息和所述预测文本内容，确定以所述易混淆文本标注信息为监督信息时的第二损失值；

基于所述第一损失值和所述第二损失值，训练所述待训练的第一文字识别网络。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括根据以下方法确定所述样本图像集中的第一样本图像以及所述第一样本图像对应的易混淆文本标注信息：

将所述样本图像集中的目标样本图像输入至预先训练的第二文字识别网络中，确定所述目标样本图像对应的预测文本内容；其中，所述第二文字识别网络是基于所述样本图像集中的样本图像和所述样本图像对应的正确文本标注信息训练得到的；

在基于所述目标样本图像的所述正确文本标注信息检测到，所述目标样本图像对应的预测文本内容中存在错误文本内容的情况下，将所述目标样本图像作为所述第一样本图像，将所述目标样本图像的预测文本内容作为所述第一样本图像的易混淆文本标注信息。

3.根据权利要求1或2所述的方法，其特征在于，所述待训练的第一文字识别网络为所述预先训练的第二文字识别网络，所述第二文字识别网络用于生成所述第一样本图像的易混淆文本标注信息。

4.根据权利要求1至3任一所述的方法，其特征在于，在所述待训练的第一文字识别网络为未经训练的初始网络的情况下，所述样本图像集中还包括，对应有正确文本标注的第二样本图像。

5.根据权利要求4所述的方法，其特征在于，所述基于所述第一损失值和所述第二损失值，训练所述待训练的第一文字识别网络，包括：

基于所述第一样本图像的第一损失值和第二损失值，确定所述第一样本图像的第一目标损失值；以及，将所述第二样本图像对应的第一损失值作为所述第二样本图像的第二目标损失值；

基于所述第一目标损失值和所述第二目标损失值，确定本次训练过程中的目标损失值；

基于所述目标损失值，训练所述待训练的第一文字识别网络。

6.根据权利要求5所述的方法，其特征在于，所述基于所述第一样本图像的第一损失值和第二损失值，确定所述第一样本图像的第一目标损失值，包括：

基于所述第一样本图像的第一损失值和所述第二损失值，确定对比损失值；所述对比损失值用于表示所述第一文字识别网络识别易混淆文本的能力；

基于所述第一样本图像的第一损失值和所述对比损失值，确定所述第一目标损失值。

7.根据权利要求6所述的方法，其特征在于，所述基于所述第一样本图像的第一损失值和所述第二损失值，确定对比损失值，包括：

将所述第一样本图像的第一损失值和所述第二损失值之差，作为所述对比损失值。

8.一种文字识别网络训练装置，其特征在于，包括：

获取模块，用于获取带有文字的样本图像集，其中，所述样本图像集中包括对应有正确文本标注信息和易混淆文本标注信息的第一样本图像；

第一确定模块，用于基于待训练的第一文字识别网络，确定所述第一样本图像对应的预测文本内容；

第二确定模块，用于基于所述正确文本标注信息和所述预测文本内容，确定以所述正确文本标注信息为监督信息时的第一损失值；以及基于所述易混淆文本标注信息和所述预测文本内容，确定以所述易混淆文本标注信息为监督信息时的第二损失值；

训练模块，用于基于所述第一损失值和所述第二损失值，训练所述待训练的第一文字识别网络。