[发明专利]文本识别方法、文本识别网络的训练方法及装置在审
申请号: | 202210034363.1 | 申请日: | 2022-01-12 |
公开(公告)号: | CN114495102A | 公开(公告)日: | 2022-05-13 |
发明(设计)人: | 张晓强;钦夏孟;章成全;姚锟 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06V30/14 | 分类号: | G06V30/14;G06V30/18;G06V30/19;G06V10/82;G06K9/62;G06N3/04;G06N3/08;G06F40/30 |
代理公司: | 北京乐知新创知识产权代理事务所(普通合伙) 11734 | 代理人: | 王曌寅 |
地址: | 100089 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 文本 识别 方法 网络 训练 装置 | ||
1.一种文本识别方法,包括:
确定待识别图像对应的编码后的序列特征;
基于所述编码后的序列特征和文本实例对应的字符特征,确定解码后的序列向量;
基于所述解码后的序列向量,确定所述待识别图像中包括的文本;
其中,所述文本实例包括预设的至少一个字符;所述字符包括字母、数字、符号、文字中至少一种。
2.根据权利要求1所述的方法,其中,所述确定待识别图像对应的编码后的序列特征包括:
将待识别图像对应的图像矩阵转换为的一维向量;
确定所述一维向量对应的特征为所述待识别图像的序列特征;
对所述序列特征进行编码,确定所述待识别图像对应的编码后的序列特征。
3.根据权利要求1所述的方法,其中,所述确定解码后的序列向量之前,所述方法还包括:
获取所述文本实例对应的嵌入值,所述嵌入值为所述文本实例对应的字符特征;
其中,不同的文本实例对应的嵌入值不同。
4.根据权利要求1所述的方法,其中,所述基于所述解码后的序列向量,确定所述待识别图像中包括的文本,包括:
将所述解码后的序列向量作为前馈子网络的输入,根据所述前馈子网络的输出,确定所述待识别图像包括的文本。
5.根据权利要求1所述的方法,其中,所述基于所述解码后的序列向量,确定所述待识别图像中包括的文本之后,所述方法还包括:
将所述解码后的序列向量与所述编码后的序列特征对应的向量相乘,得到乘积结果;
基于所述乘积结果,确定所述文本在所述编码后的序列特征中的位置信息;
基于所述文本在所述编码后的序列特征中的位置信息,确定所述文本在所述待识别图像的位置信息;
基于所述文本在所述待识别图像的位置信息和所述待识别图像中包括的文本,按照所述文本在所述待识别图像中的顺序,输出所述待识别图像中包括的文本。
6.一种文本识别网络的训练方法,其中,文本识别网络包括编码子网络、译码子网络和输出子网络,所述方法包括:
基于所述编码子网络确定训练样本集中样本图像对应的编码后的序列样本特征;
以所述编码后的序列样本特征和文本实例样本对应的字符样本特征作为所述译码子网络的跨层注意力层的输入,将所述跨层注意力层的输出确定为解码后的样本序列向量;
将所述解码后的样本序列向量作为所述输出子网络的输入,根据所述输出子网络的输出,确定所述样本图像中包括的预测文本;
匹配所述样本图像包括的预测文本和所述样本图像包括的标注文本,基于匹配结果调整所述文本识别网络的参数;
其中,所述文本实例包括预设的至少一个字符;所述字符包括字母、数字、符号、文字中至少一种。
7.根据权利要求6所述的方法,其中,所述文本识别网络还包括基础子网络;所述基于所述编码子网络确定训练样本集中样本图像对应的编码后的序列样本特征,包括:
所述基础子网络将所述样本图像对应的图像矩阵转换为表征所述样本图像的一维向量;
所述样本图像的一维向量对应的特征为所述样本图像的序列样本特征;
所述编码子网络对所述序列样本特征进行编码,确定训练样本集中样本图像对应的编码后的序列样本特征。
8.根据权利要求6所述的方法,其中,所述确定解码后的样本序列向量之前,所述方法还包括:
将所述文本实例样本输入至所述译码子网络的自注意力层,基于所述自注意力层的输出获取所述文本实例样本对应的嵌入值,所述嵌入值为所述文本实例样本对应的字符样本特征;
其中,不同的文本实例样本对应的嵌入值不同。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202210034363.1/1.html,转载请声明来源钻瓜专利网。