[发明专利]一种模型训练的方法以及装置

说明书

本说明书公开了一种模型训练的方法以及装置，获取各待标注图像，针对每个待标注图像，将该待标注图像输入到待训练的文字行识别模型中，得到文字行识别结果，以及将该待标注图像输入到预设的单字识别模型中，得到单字识别结果。而后，根据单字识别结果，对文字行识别结果进行校正，得到校正后文字行识别结果。最后，根据校正后文字行识别结果，对该待标注图像进行标注，得到标注后图像，通过标注后图像，对待训练的文字行识别模型进行模型训练。本方法将单字识别结果和文字行识别结果进行结合，可以有效的提高数据标注的准确性，从而使得文字行识别模型的准确率提高。

技术领域

本说明书涉及计算机技术领域，尤其涉及一种模型训练的方法以及装置。

背景技术

随着电子技术和网络技术的不断发展，文字识别算法得到了巨大进展。目前，训练一个高精度的文字识别模型需要大量精细的人工标注数据作为监督，而准备一个满足需求的标注数据集对于时间成本和人力成本的耗费十分巨大。

在现有技术中，将无标注的数据输入到两个不同的文字识别模型中，输出两个识别结果，通过一个判别器将其中一个识别结果作为数据标注，再使用标注后的数据对文字识别模型进行训练，而判别器无法准确的选择出正确的识别结果，所以，当这两个文字识别模型输出的识别结果存在错误的情况下，将导致数据的标注出现错误，从而使得文字识别模型的准确率降低。

因此，如何提高对数据标注的准确性，从而使得文字识别模型的准确率提高，则是一个亟待解决的问题。

发明内容

本说明书提供一种模型训练的方法、装置、存储介质及电子设备，以部分的解决现有技术存在的上述问题。

本说明书采用下述技术方案：

本说明书提供了一种模型训练的方法，包括：

获取各待标注图像；

针对每个待标注图像，将该待标注图像输入到待训练的文字行识别模型中，得到文字行识别结果，以及将该待标注图像输入到预设的单字识别模型中，得到单字识别结果；

根据所述单字识别结果，对所述文字行识别结果进行校正，得到校正后文字行识别结果；

根据所述校正后文字行识别结果，对该待标注图像进行标注，得到标注后图像；

通过所述标注后图像，对所述待训练的文字行识别模型进行模型训练。

可选地，将该待标注图像输入到待训练的文字行识别模型中，得到文字行识别结果，具体包括：

将该待标注图像输入到待训练的文字行识别模型中，得到所述文字行识别模型识别出的该待标注图像中所包含的文字行信息，以及所述文字行信息在该待标注图像中的文字行位置；

将所述文字行位置以及所述文字行信息，作为所述文字行识别结果；

将该待标注图像输入到预设的单字识别模型中，得到单字识别结果，具体包括：

将该待标注图像输入到所述单字识别模型中，得到所述单字识别模型识别出的该待标注图像中所包含的各单字信息，以及每个单字信息在该待标注图像中的单字位置；

将所述各单字信息以及每个单字信息在该待标注图像中的单字位置，作为所述单字识别结果。

可选地，根据所述单字识别结果，对所述文字行识别结果进行校正之前，所述方法还包括：

针对所述单字识别结果中的每个单字信息，根据确定出的该单字信息在该待标注图像中的单字位置以及所述文字行位置，确定该单字信息与所述文字行信息在该待标注图像中的区域重合度；

若确定所述区域重合度不小于设定重合度，确定该单字信息归属于所述文字行信息；