[发明专利]图像字符识别模型训练方法、图像字符识别方法及装置

说明书

本申请实施例公开了图像字符识别模型训练方法、图像字符识别方法及装置，利用训练图像和训练图像对应的字符标识，训练由第一模型和第二模型构成的图像字符识别模型。训练图像包括遮蔽字符区域和显示字符区域。利用包括遮蔽字符区域的训练图像对图像字符识别模型进行训练，能够使得图像字符识别模型更好地提取双向上下文信息，如此训练得到的图像字符识别模型准确率较高。利用训练完成的图像字符识别模型对待识别图像进行识别，可以得到更为准确的待识别图像包括的字符。

技术领域

本申请涉及图像处理领域，具体涉及一种图像字符识别模型训练方法、图像字符识别方法及装置。

背景技术

OCR(Optical Character Recognition，光学字符识别)是对包括字符的图像进行识别分析，得到图像中的字符的技术。利用OCR技术，能够获取图像中的字符信息。

采用OCR技术可以先确定图像中的字符区域，再对图像中的字符区域进行分割，最后对字符区域中的字符进行识别。目前，利用OCR技术识别对图像中的字符的准确度较低。

发明内容

有鉴于此，本申请实施例提供一种图像字符识别模型训练方法、图像字符识别方法及装置，能够较为准确地对图像中的字符进行识别。

基于此，本申请实施例提供的技术方案如下：

第一方面，本申请实施例提供一种图像字符识别模型训练方法，所述方法包括：

将训练图像输入第一模型，得到所述第一模型输出的显示字符对应的第一特征向量和遮蔽字符对应的第二特征向量，所述训练图像包括至少一个显示字符区域和至少一个遮蔽字符区域，所述显示字符区域包括至少一个所述显示字符，所述所述遮蔽字符区域用于遮蔽至少一个所述遮蔽字符；

将所述遮蔽字符的第二特征向量输入第二模型，得到所述第二模型输出的遮蔽字符的预测字符；

根据与所述训练图像对应的字符标识和所述遮蔽字符区域的预测字符，训练图像字符识别模型，所述图像字符识别模型包括所述第一模型和所述第二模型，所述字符标识用于标识所述遮蔽字符。

第二方面，本申请实施例提供一种图像字符识别方法，所述方法包括：

将待识别图像输入图像字符识别模型，所述待识别图像包括至少一个待识别字符，所述图像字符识别模型是基于上述第一方面所述的图像字符识别模型的训练方法训练得到的；

获取所述图像字符识别模型输出的识别结果。

第三方面，本申请实施例提供一种图像字符识别模型训练装置，所述装置包括：

第一输入单元，用于将训练图像输入第一模型，得到所述第一模型输出的显示字符对应的第一特征向量和遮蔽字符对应的第二特征向量，所述训练图像包括至少一个显示字符区域和至少一个遮蔽字符区域，所述显示字符区域包括至少一个所述显示字符，所述所述遮蔽字符区域用于遮蔽至少一个所述遮蔽字符；

第二输入单元，用于将所述遮蔽字符的第二特征向量输入第二模型，得到所述第二模型输出的遮蔽字符的预测字符；

训练单元，用于根据与所述训练图像对应的字符标识和所述遮蔽字符区域的预测字符，训练图像字符识别模型，所述图像字符识别模型包括所述第一模型和所述第二模型，所述字符标识用于标识所述遮蔽字符。

第四方面，本申请实施例提供一种图像字符识别装置，所述装置包括：

输入单元，用于将待识别图像输入图像字符识别模型，所述待识别图像包括至少一个待识别字符，所述图像字符识别模型是基于上述第一方面所述的图像字符识别模型的训练方法训练得到的；

识别单元，用于获取所述图像字符识别模型输出的识别结果。

第五方面，本申请实施例提供一种电子设备，包括：