一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

文本识别方法、装置、可读介质及电子设备与流程

2022-03-23 06:40:17 来源:中国专利 TAG:


1.本公开涉及图像处理技术领域,具体地,涉及一种文本识别方法、装置、可读介质及电子设备。


背景技术:

2.随着文本图像识别技术的广泛应用,人们对文本图像识别的准确率的要求越来越高,需要能够准确识别文本图像中的每个字符。相关技术中,通过基于深度学习的神经网络模型识别文本图像中的文本内容。
3.但是,针对部分比较复杂的文本图像,例如广告、电影海报等文本图像,文本图像中的标题往往会使用特殊排版,这样,通过同一个神经网络模型对该文本图像进行识别时,会导致部分字符无法正常识别,从而使得文本图像识别的准确率比较低。


技术实现要素:

4.提供该发明内容部分以便以简要的形式介绍构思,这些构思将在后面的具体实施方式部分被详细描述。该发明内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征,也不旨在用于限制所要求的保护的技术方案的范围。
5.第一方面,本公开提供一种文本识别方法,所述方法包括:
6.获取待识别的文本图像对应的文本特征图像;
7.根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;
8.根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。
9.第二方面,本公开提供一种文本识别装置,所述装置包括:
10.第一特征图像获取模块,用于获取待识别的文本图像对应的文本特征图像;
11.第二特征图像获取模块,用于根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;
12.文本内容获取模块,用于根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。
13.第三方面,本公开提供一种计算机可读介质,其上存储有计算机程序,该程序被处理装置执行时实现本公开第一方面所述方法的步骤。
14.第四方面,本公开提供一种电子设备,包括:
15.存储装置,其上存储有计算机程序;
16.处理装置,用于执行所述存储装置中的所述计算机程序,以实现本公开第一方面所述方法的步骤。
17.通过上述技术方案,通过获取待识别的文本图像对应的文本特征图像;根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。也就是说,本公开对该文本图像的维度进行拆分,可以根据表征该文本图像的高度方向特征的高度特征图像、表征该文本图像的宽度方向特征的宽度特征图像以及该文本图像对应的文本特征图像,获取该文本图像对应的文本内容,这样,在文本图像识别过程中,可以结合该文本图像的不同维度的特征,从而提高了文本图像识别的准确率。
18.本公开的其他特征和优点将在随后的具体实施方式部分予以详细说明。
附图说明
19.结合附图并参考以下具体实施方式,本公开各实施例的上述和其他特征、优点及方面将变得更加明显。贯穿附图中,相同或相似的附图标记表示相同或相似的元素。应当理解附图是示意性的,原件和元素不一定按照比例绘制。在附图中:
20.图1是根据一示例性实施例提供的一种文本识别方法的流程图;
21.图2是根据一示例性实施例提供的另一种文本识别方法的流程图;
22.图3是根据一示例性实施例提供的一种字符识别方法的流程图;
23.图4是根据一示例性实施例提供的一种缩放点乘积注意力子层的示意图;
24.图5是根据一示例性实施例提供的一种文本识别方法的示意图;
25.图6是根据一示例性实施例提供的一种文本识别装置的框图;
26.图7是根据一示例性实施例提供的另一种文本识别装置的框图;
27.图8是根据一示例性实施例提供的一种电子设备的框图。
具体实施方式
28.下面将参照附图更详细地描述本公开的实施例。虽然附图中显示了本公开的某些实施例,然而应当理解的是,本公开可以通过各种形式来实现,而且不应该被解释为限于这里阐述的实施例,相反提供这些实施例是为了更加透彻和完整地理解本公开。应当理解的是,本公开的附图及实施例仅用于示例性作用,并非用于限制本公开的保护范围。
29.应当理解,本公开的方法实施方式中记载的各个步骤可以按照不同的顺序执行,和/或并行执行。此外,方法实施方式可以包括附加的步骤和/或省略执行示出的步骤。本公开的范围在此方面不受限制。
30.本文使用的术语“包括”及其变形是开放性包括,即“包括但不限于”。术语“基于”是“至少部分地基于”。术语“一个实施例”表示“至少一个实施例”;术语“另一实施例”表示“至少一个另外的实施例”;术语“一些实施例”表示“至少一些实施例”。其他术语的相关定义将在下文描述中给出。
31.需要注意,本公开中提及的“第一”、“第二”等概念仅用于对不同的装置、模块或单元进行区分,并非用于限定这些装置、模块或单元所执行的功能的顺序或者相互依存关系。
32.需要注意,本公开中提及的“一个”、“多个”的修饰是示意性而非限制性的,本领域
技术人员应当理解,除非在上下文另有明确指出,否则应该理解为“一个或多个”。
33.本公开实施方式中的多个装置之间所交互的消息或者信息的名称仅用于说明性的目的,而并不是用于对这些消息或信息的范围进行限制。
34.首先,对本公开的应用场景进行说明。传统的ocr(optical character recognition,光学字符识别)技术分为文本检测和文本识别两个步骤,通过文本检测对文本图像进行分割得到该文本图像中的文字区域,通过文本识别获取该文字区域中的文本内容。相关技术中,基于深度学习的文本识别主要包括以crnn(convolutional recurrent neural network,卷积循环神经网络)为代表的ctc(connectionist temporal classification,联结主义时间分类)方法和以transformer为代表的attention(注意力)方法,ctc方法和attention方法对于水平文本图像的识别能力较强,但是针对部分特殊文本图像,例如广告、电影海报等文本图像,由于文本图像的标题往往使用了特殊排版,其中包括了不同高度的文本行,全部文本图像均按照一维文本行进行下采样处理,会使得二维文本的高度被过度压缩,导致信息丢失,从而无法正常识别;全部文本图像均按照二维文本的高度进行下采样处理,使得需要处理的序列长度更长,计算量大幅增加,导致文本识别的效率太低。
35.为了技术上述存在的技术问题,本公开提供一种文本识别方法、装置、可读介质及电子设备,通过对该文本图像的维度进行拆分,可以根据表征该文本图像的高度方向特征的高度特征图像、表征该文本图像的宽度方向特征的宽度特征图像以及该文本图像对应的文本特征图像,获取该文本图像对应的文本内容,这样,在文本图像识别过程中,可以结合该文本图像的不同维度的特征,从而提高了文本图像识别的准确率。
36.下面结合具体实施例对本公开进行说明。
37.图1是根据一示例性实施例提供的一种文本识别方法的流程图,如图1所示,该方法可以包括:
38.s101、获取待识别的文本图像对应的文本特征图像。
39.在本步骤中,在获取待识别的文本图像后,可以对该文本图像进行下采样处理,得到该文本图像对应的文本特征图像。示例地,可以通过预先训练的文本特征图像获取模型对该文本图像进行下采样处理,得到该文本图像对应的文本特征图像。其中,该文本特征图像获取模型可以是基于cnn(convolutional neural networks,卷积神经网络模型)框架,通过现有技术的模型训练方法训练得到的,此处不再赘述。其中,该cnn框架可以包括resnet18的前两个block,该文本特征图像获取模型进行下采样处理的下采样倍率可以通过试验预先测试得到,例如,该下采样倍率可以是4。若该文本图像对应的尺寸为(h,w),该文本特征图像获取模型进行下采样处理的下采样倍率为n,则对该文本图像进行下采样处理后得到的文本特征图像的尺寸为(h/n,w/n)。
40.s102、根据该文本特征图像,确定该文本图像对应的高度特征图像和宽度特征图像。
41.其中,该高度特征图像用于表征该文本图像的高度方向的特征,该宽度特征图像用于表征该文本图像的宽度方向的特征。
42.在本步骤中,在得到该文本图像对应的文本特征图像后,可以将该文本特征图像输入预先训练的高度特征获取模型,以获取该高度特征获取模型输出的该文本图像对应的
该高度特征图像,将该文本特征图像输入预先训练的宽度特征获取模型,以获取该宽度特征获取模型输出的该文本图像对应的该宽度特征图像。
43.s103、根据该文本特征图像、该高度特征图像以及该宽度特征图像,通过预先训练的文本识别模型,获取该文本图像对应的文本内容。
44.其中,该文本识别模型可以包括第一编码子模型、第二编码子模型以及字符识别子模型,该文本识别模型的训练方法可以参考现有技术的模型训练方法,此处不再赘述。
45.在本步骤中,在得到该文本特征图像、该高度特征图像以及该宽度特征图像后,可以将该文本特征图像、该高度特征图像以及该宽度特征图像输入该文本识别模型,以获取该文本识别模型输出的该文本图像对应的文本内容。
46.采用上述方法,通过对该文本图像的维度进行拆分,可以根据表征该文本图像的高度方向特征的高度特征图像、表征该文本图像的宽度方向特征的宽度特征图像以及该文本图像对应的文本特征图像,获取该文本图像对应的文本内容,这样,在文本图像识别过程中,可以结合该文本图像的不同维度的特征,从而提高了文本图像识别的准确率。
47.图2是根据一示例性实施例提供的另一种文本识别方法的流程图,如图2所示,该方法可以包括:
48.s201、获取待识别的文本图像对应的文本特征图像。
49.s202、将该文本特征图像输入预先训练的高度特征获取模型,以获取该高度特征获取模型输出的该文本图像对应的高度特征图像。
50.其中,该高度特征图像用于表征该文本图像的高度方向的特征,该高度特征获取模型可以基于reduction(降维)模型,通过现有技术的模型训练方法进行训练得到,此处不再赘述。
51.在本步骤中,在得到该文本特征图像后,可以将该文本特征图像输入该高度特征获取模型,通过该高度特征获取模型对该文本特征图像进行处理,得到该文本图像对应的高度特征图像,该高度特征图像的宽度为1,即该高度特征图像仅体现该文本图像的高度方向的特征。
52.s203、将该文本特征图像输入预先训练的宽度特征获取模型,以获取该宽度特征获取模型输出的该文本图像对应的宽度特征图像。
53.其中,该宽度特征图像用于表征该文本图像的宽度方向的特征,该宽度特征获取模型也可以基于reduction模型,通过现有技术的模型训练方法进行训练得到,此处不再赘述。该宽度特征获取模型的参数与该高度特征获取模型的参数不同。
54.在本步骤中,在得到该文本特征图像后,可以将该文本特征图像输入该宽度特征获取模型,通过该宽度特征获取模型对该文本特征图像进行处理,得到该文本图像对应的宽度特征图像,该宽度特征图像的高度为1,即该宽度特征图像仅体现该文本图像的宽度方向的特征。
55.需要说明的是,本公开对上述步骤s202和步骤s203的执行顺序不作限定,在得到该文本图像对应的文本特征图像后,可以先执行步骤s202,再执行步骤s203,也可以先执行步骤s203,再执行步骤s202,还可以并行同步执行步骤s202和步骤s203。
56.s204、将该高度特征图像输入第一编码子模型,以获取该第一编码子模型输出的该高度特征图像对应的高度编码特征向量。
57.其中,该第一编码子模型的结构可以与现有技术的transformer模型中的编码器的结构相同。
58.在本步骤中,在得到该高度特征图像后,可以将该高度特征图像输入该第一编码子模型,通过该第一编码子模型对该高度特征图像进行处理,得到该高度特征图像对应的高度编码特征向量。
59.s205、将该宽度特征图像输入第二编码子模型,以获取该第二编码子模型输出的该宽度特征图像对应的宽度编码特征向量。
60.其中,该第二编码子模型的结构也可以与现有技术的transformer模型中的编码器的结构相同,该第二编码子模型的参数与该第一编码子模型的参数不同。
61.在本步骤中,在得到该宽度特征图像后,可以将该宽度特征图像输入该第二编码子模型,通过该第二编码子模型对该宽度特征图像进行处理,得到该宽度特征图像对应的宽度编码特征向量。
62.s206、根据该文本特征图像、该高度编码特征向量以及该宽度编码特征向量,通过字符识别子模型,获取该文本图像对应的文本内容。
63.在本步骤中,在得到该文本特征图像、该高度编码特征向量以及该宽度编码特征向量后,可以循环执行字符识别步骤,直至该字符识别子模型输出的目标字符为预设终止字符,将新的文本字符作为该文本图像对应的文本内容。示例地,在该字符识别子模型输出的该目标字符为“/eos”的情况下,若该文本字符包括“o”、“c”、“r”、
“”
、“i”、“s”、
“”
、“g”、“r”、“e”、“a”、“t”、“!”,则该文本图像对应的文本内容可以是“ocr is great!”64.其中,图3是根据一示例性实施例提供的一种字符识别方法的流程图,如图3所示,该字符识别步骤可以包括:
65.s1、获取该文本图像中已识别的文本字符对应的字符嵌入特征向量。
66.该文本图像中已识别的文本字符可以是该字符识别模型输出的所有目标字符,在得到该文本字符后,可以通过现有技术的字符嵌入模型(例如,transformer模型中的输出嵌入),获取该文本字符对应的字符嵌入特征向量。
67.s2、将该文本特征图像、该高度编码特征向量、该宽度编码特征向量以及该字符嵌入特征向量输入该字符识别子模型,以获取该字符识别子模型输出的该目标字符。
68.其中,该字符识别子模型可以包括第一解码子模型和第二解码子模型,该第一解码子模型可以包括多头注意力层、残差链接&正则化层(add&norm),该多头注意力层的输出端与该add&norm的输入端耦合,该add&norm的输出端与该第二解码子模型的输入端耦合,该第二解码子模型可以包括多头注意力层和前馈神经网络层,该多头注意力层可以包括第一线性子层、缩放点乘积注意力子层、连接子层以及第二线性子层,该第一线性子层的输出端与该缩放点乘积注意力子层的输入端耦合,该缩放点乘积注意力子层的输出端与该连接子层的输入端耦合,该连接子层的输出端与该第二线性子层的输入端耦合。
69.在得到该文本特征图像、该高度编码特征向量、该宽度编码特征向量以及该字符嵌入特征向量后,可以将该字符嵌入特征向量输入该第一解码子模型,以获取该第一解码子模型输出的该文本字符对应的字符解码特征向量。
70.进一步地,在得到该文本字符对应的字符解码特征向量后,可以将该文本特征图像、该高度编码特征向量、该宽度编码特征向量以及该字符解码特征向量输入该第二解码
子模型,以获取该第二解码子模型输出的该目标字符。
71.在一种可能的实现方式中,可以将该高度编码特征向量、该宽度编码特征向量以及该字符解码特征向量输入该第一线性子层,以获取该第一线性子层输出的目标高度编码特征向量、目标宽度编码特征向量以及目标字符解码特征向量;将该文本特征图像、该目标高度编码特征向量、该目标宽度编码特征向量以及该目标字符解码特征向量输入该缩放点乘积注意力子层,以通过该缩放点乘积注意力子层根据该文本特征图像、该目标高度编码特征向量、该目标宽度编码特征向量以及该目标字符解码特征向量,确定文本解码特征向量;将该文本解码特征向量输入该连接子层,以通过该连接子层和该第二线性子层输出该目标字符。
72.其中,该缩放点乘积注意力子层可以根据该目标高度编码特征向量和该目标字符解码特征向量,确定高度解码特征向量,根据该目标宽度编码特征向量和该目标字符解码特征向量,确定宽度解码特征向量,根据该高度解码特征向量和该宽度解码特征向量,确定待定解码特征向量,并根据该待定解码特征向量和该文本特征图像,确定该文本解码特征向量。图4是根据一示例性实施例提供的一种缩放点乘积注意力子层的示意图,如图4所示,该缩放点乘积注意力子层包括第一缩放点乘积注意力模块和第二缩放点乘积注意力模块,该第一缩放点乘积注意力模块和该第二缩放点乘积注意力模块的结构相同,均包括矩阵相乘(matmul)、放缩处理(scale)、掩膜过滤(mask(opt.))以及分类处理(softmax)。其中,q为该目标字符解码特征向量,fw为该目标宽度编码特征向量,fh为该目标高度编码特征向量,f
2d
为该文本特征图像。通过该第一缩放点乘积注意力模块对输入的该目标高度编码特征向量和该目标字符解码特征向量进行处理,得到该高度解码特征向量,通过该第二缩放点乘积注意力模块对输入的该目标宽度编码特征向量和该目标字符解码特征向量进行处理,得到该宽度解码特征向量,之后,通过torch.sum()函数确定该高度解码特征向量与该宽度解码特征向量的和值,将该和值作为该待定解码特征向量,最后,通过matmul对该待定解码特征向量和该文本特征图像进行矩阵相乘处理,对该待定解码特征向量的维度进行扩展,得到该文本解码特征向量。
73.需要说明的是,该第一缩放点乘积注意力模块和该第二缩放点乘积注意力模块的处理流程,可以参考现有技术的transformer模型中的多头注意力层的处理流程,此处不再赘述。
74.s3、将该文本字符和该目标字符作为新的文本字符。
75.在该字符识别子模型输出该目标字符后,可以对该文本字符进行更新,将该文本字符和该目标字符作为新的文本字符。
76.图5是根据一示例性实施例提供的一种文本识别方法的示意图,如图5所示,输入的文本图像的尺寸为(h,w),通过文本特征图像获取模型(图中的cnn)进行下采样处理后,得到尺寸为(h/4,w/4)的文本特征图像f
2d
(下采样倍率为4),将该文本特征图像分别输入高度特征获取模型和宽度特征获取模型,图中的两个降维模型(reduction),,得到尺寸为(h/4,1)高度特征图像和尺寸为(1,w/4)宽度特征图像,之后,将该高度特征图像和该宽度特征图像分别输入第一编码子模型和第二编码子模型,得到该高度特征图像对应的高度编码特征向量fh和该宽度特征图像对应的宽度编码特征向量fw,最后,将该高度编码特征向量fh、该宽度编码特征向量fw以及该文本特征图像f
2d
输入第二解码子模型的多头注意力层,最
后,通过该字符识别子模型得到输出该目标字符。
77.采用上述方法,通过对该文本图像的维度进行拆分,可以根据表征该文本图像的高度方向特征的高度特征图像、表征该文本图像的宽度方向特征的宽度特征图像以及该文本图像对应的文本特征图像,获取该文本图像对应的文本内容,这样,在文本图像识别过程中,可以结合该文本图像的不同维度的特征,从而提高了文本图像识别的准确率;另外,相比该文本特征图像,该高度特征图像和该宽度特征图像的维度比较低,降低了编码处理过程中的计算量,提高了文本图像识别的效率,并且,第二解码子模型最终的定位仍然处于和文本特征图像同等分辨率的粒度,确保了文本图像识别的准确率。
78.图6是根据一示例性实施例提供的一种文本识别装置的框图,如图6所示,该装置可以包括:
79.第一特征图像获取模块601,用于获取待识别的文本图像对应的文本特征图像;
80.第二特征图像获取模块602,用于根据该文本特征图像,确定该文本图像对应的高度特征图像和宽度特征图像,该高度特征图像用于表征该文本图像的高度方向的特征,该宽度特征图像用于表征该文本图像的宽度方向的特征;
81.文本内容获取模块603,用于根据该文本特征图像、该高度特征图像以及该宽度特征图像,通过预先训练的文本识别模型,获取该文本图像对应的文本内容。
82.可选地,该第二特征图像获取模块602,还用于:
83.将该文本特征图像输入预先训练的高度特征获取模型,以获取该高度特征获取模型输出的该文本图像对应的该高度特征图像;
84.将该文本特征图像输入预先训练的宽度特征获取模型,以获取该宽度特征获取模型输出的该文本图像对应的该宽度特征图像。
85.可选地,该文本识别模型包括第一编码子模型、第二编码子模型以及字符识别子模型;图7是根据一示例性实施例提供的另一种文本识别装置的框图,如图7所示,该装置还包括:
86.高度编码特征向量获取模块604,用于将该高度特征图像输入该第一编码子模型,以获取该第一编码子模型输出的该高度特征图像对应的高度编码特征向量;
87.宽度编码特征向量获取模块605,用于将该宽度特征图像输入该第二编码子模型,以获取该第二编码子模型输出的该宽度特征图像对应的宽度编码特征向量;
88.该文本内容获取模块603,还用于:
89.根据该文本特征图像、该高度编码特征向量以及该宽度编码特征向量,通过该字符识别子模型,获取该文本图像对应的文本内容。
90.可选地,该文本内容获取模块603,还用于:
91.循环执行字符识别步骤,直至该字符识别子模型输出的目标字符为预设终止字符,将新的文本字符作为该文本图像对应的文本内容;该字符识别步骤包括:
92.获取该文本图像中已识别的文本字符对应的字符嵌入特征向量;
93.将该文本特征图像、该高度编码特征向量、该宽度编码特征向量以及该字符嵌入特征向量输入该字符识别子模型,以获取该字符识别子模型输出的该目标字符;
94.将该文本字符和该目标字符作为新的文本字符。
95.可选地,该字符识别子模型包括第一解码子模型和第二解码子模型;该文本内容
获取模块603,还用于:
96.将该字符嵌入特征向量输入该第一解码子模型,以获取该第一解码子模型输出的该文本字符对应的字符解码特征向量;
97.将该文本特征图像、该高度编码特征向量、该宽度编码特征向量以及该字符解码特征向量输入该第二解码子模型,以获取该第二解码子模型输出的该目标字符。
98.可选地,该第二解码子模型包括多头注意力层和前馈神经网络层,该多头注意力层包括第一线性子层、缩放点乘积注意力子层、连接子层以及第二线性子层,该第一线性子层的输出端与该缩放点乘积注意力子层的输入端耦合,该缩放点乘积注意力子层的输出端与该连接子层的输入端耦合,该连接子层的输出端与该第二线性子层的输入端耦合;该文本内容获取模块603,还用于:
99.将该高度编码特征向量、该宽度编码特征向量以及该字符解码特征向量输入该第一线性子层,以获取该第一线性子层输出的目标高度编码特征向量、目标宽度编码特征向量以及目标字符解码特征向量;
100.将该文本特征图像、该目标高度编码特征向量、该目标宽度编码特征向量以及该目标字符解码特征向量输入该缩放点乘积注意力子层,以通过该缩放点乘积注意力子层根据该文本特征图像、该目标高度编码特征向量、该目标宽度编码特征向量以及该目标字符解码特征向量,确定文本解码特征向量;
101.将该文本解码特征向量输入该连接子层,以通过该连接子层和该第二线性子层输出该目标字符。
102.可选地,该文本内容获取模块603,还用于:
103.根据该目标高度编码特征向量和该目标字符解码特征向量,确定高度解码特征向量;
104.根据该目标宽度编码特征向量和该目标字符解码特征向量,确定宽度解码特征向量;
105.根据该高度解码特征向量和该宽度解码特征向量,确定待定解码特征向量;
106.根据该待定解码特征向量和该文本特征图像,确定该文本解码特征向量。
107.通过上述装置,通过对该文本图像的维度进行拆分,可以根据表征该文本图像的高度方向特征的高度特征图像、表征该文本图像的宽度方向特征的宽度特征图像以及该文本图像对应的文本特征图像,获取该文本图像对应的文本内容,这样,在文本图像识别过程中,可以结合该文本图像的不同维度的特征,从而提高了文本图像识别的准确率。
108.关于上述实施例中的装置,其中各个模块执行操作的具体方式已经在有关该方法的实施例中进行了详细描述,此处将不做详细阐述说明。
109.下面参考图8,其示出了适于用来实现本公开实施例的电子设备(例如终端设备或服务器)800的结构示意图。本公开实施例中的终端设备可以包括但不限于诸如移动电话、笔记本电脑、数字广播接收器、pda(个人数字助理)、pad(平板电脑)、pmp(便携式多媒体播放器)、车载终端(例如车载导航终端)等等的移动终端以及诸如数字tv、台式计算机等等的固定终端。图8示出的电子设备仅仅是一个示例,不应对本公开实施例的功能和使用范围带来任何限制。
110.如图8所示,电子设备800可以包括处理装置(例如中央处理器、图形处理器等)
801,其可以根据存储在只读存储器(rom)802中的程序或者从存储装置808加载到随机访问存储器(ram)803中的程序而执行各种适当的动作和处理。在ram 803中,还存储有电子设备800操作所需的各种程序和数据。处理装置801、rom 802以及ram 803通过总线804彼此相连。输入/输出(i/o)接口805也连接至总线804。
111.通常,以下装置可以连接至i/o接口805:包括例如触摸屏、触摸板、键盘、鼠标、摄像头、麦克风、加速度计、陀螺仪等的输入装置806;包括例如液晶显示器(lcd)、扬声器、振动器等的输出装置807;包括例如磁带、硬盘等的存储装置808;以及通信装置809。通信装置809可以允许电子设备800与其他设备进行无线或有线通信以交换数据。虽然图8示出了具有各种装置的电子设备800,但是应理解的是,并不要求实施或具备所有示出的装置。可以替代地实施或具备更多或更少的装置。
112.特别地,根据本公开的实施例,上文参考流程图描述的过程可以被实现为计算机软件程序。例如,本公开的实施例包括一种计算机程序产品,其包括承载在非暂态计算机可读介质上的计算机程序,该计算机程序包含用于执行流程图所示的方法的程序代码。在这样的实施例中,该计算机程序可以通过通信装置809从网络上被下载和安装,或者从存储装置808被安装,或者从rom 802被安装。在该计算机程序被处理装置801执行时,执行本公开实施例的方法中限定的上述功能。
113.需要说明的是,本公开上述的计算机可读介质可以是计算机可读信号介质或者计算机可读存储介质或者是上述两者的任意组合。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。计算机可读存储介质的更具体的例子可以包括但不限于:具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机访问存储器(ram)、只读存储器(rom)、可擦式可编程只读存储器(eprom或闪存)、光纤、便携式紧凑磁盘只读存储器(cd-rom)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本公开中,计算机可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。而在本公开中,计算机可读信号介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了计算机可读的程序代码。这种传播的数据信号可以采用多种形式,包括但不限于电磁信号、光信号或上述的任意合适的组合。计算机可读信号介质还可以是计算机可读存储介质以外的任何计算机可读介质,该计算机可读信号介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。计算机可读介质上包含的程序代码可以用任何适当的介质传输,包括但不限于:电线、光缆、rf(射频)等等,或者上述的任意合适的组合。
114.在一些实施方式中,客户端、服务器可以利用诸如http(hypertext transfer protocol,超文本传输协议)之类的任何当前已知或未来研发的网络协议进行通信,并且可以与任意形式或介质的数字数据通信(例如,通信网络)互连。通信网络的示例包括局域网(“lan”),广域网(“wan”),网际网(例如,互联网)以及端对端网络(例如,ad hoc端对端网络),以及任何当前已知或未来研发的网络。
115.上述计算机可读介质可以是上述电子设备中所包含的;也可以是单独存在,而未装配入该电子设备中。
116.上述计算机可读介质承载有一个或者多个程序,当上述一个或者多个程序被该电
子设备执行时,使得该电子设备:获取待识别的文本图像对应的文本特征图像;根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。
117.可以以一种或多种程序设计语言或其组合来编写用于执行本公开的操作的计算机程序代码,上述程序设计语言包括但不限于面向对象的程序设计语言—诸如java、smalltalk、c ,还包括常规的过程式程序设计语言——诸如“c”语言或类似的程序设计语言。程序代码可以完全地在用户计算机上执行、部分地在用户计算机上执行、作为一个独立的软件包执行、部分在用户计算机上部分在远程计算机上执行、或者完全在远程计算机或服务器上执行。在涉及远程计算机的情形中,远程计算机可以通过任意种类的网络——包括局域网(lan)或广域网(wan)——连接到用户计算机,或者,可以连接到外部计算机(例如利用因特网服务提供商来通过因特网连接)。
118.附图中的流程图和框图,图示了按照本公开各种实施例的系统、方法和计算机程序产品的可能实现的体系架构、功能和操作。在这点上,流程图或框图中的每个方框可以代表一个模块、程序段、或代码的一部分,该模块、程序段、或代码的一部分包含一个或多个用于实现规定的逻辑功能的可执行指令。也应当注意,在有些作为替换的实现中,方框中所标注的功能也可以以不同于附图中所标注的顺序发生。例如,两个接连地表示的方框实际上可以基本并行地执行,它们有时也可以按相反的顺序执行,这依所涉及的功能而定。也要注意的是,框图和/或流程图中的每个方框、以及框图和/或流程图中的方框的组合,可以用执行规定的功能或操作的专用的基于硬件的系统来实现,或者可以用专用硬件与计算机指令的组合来实现。
119.描述于本公开实施例中所涉及到的模块可以通过软件的方式实现,也可以通过硬件的方式来实现。其中,模块的名称在某种情况下并不构成对该模块本身的限定,例如,第一特征图像获取模块还可以被描述为“获取待识别的文本图像对应的文本特征图像的模块”。
120.本文中以上描述的功能可以至少部分地由一个或多个硬件逻辑部件来执行。例如,非限制性地,可以使用的示范类型的硬件逻辑部件包括:现场可编程门阵列(fpga)、专用集成电路(asic)、专用标准产品(assp)、片上系统(soc)、复杂可编程逻辑设备(cpld)等等。
121.在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行系统、装置或设备使用或与指令执行系统、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体系统、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(ram)、只读存储器(rom)、可擦除可编程只读存储器(eprom或快闪存储器)、光纤、便捷式紧凑盘只读存储器(cd-rom)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
122.根据本公开的一个或多个实施例,示例1提供了一种文本识别方法,所述方法包
括:获取待识别的文本图像对应的文本特征图像;根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。
123.根据本公开的一个或多个实施例,示例2提供了示例1的方法,所述根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像包括:将所述文本特征图像输入预先训练的高度特征获取模型,以获取所述高度特征获取模型输出的所述文本图像对应的所述高度特征图像;将所述文本特征图像输入预先训练的宽度特征获取模型,以获取所述宽度特征获取模型输出的所述文本图像对应的所述宽度特征图像。
124.根据本公开的一个或多个实施例,示例3提供了示例1的方法,所述文本识别模型包括第一编码子模型、第二编码子模型以及字符识别子模型;在所述根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容前,所述方法还包括:将所述高度特征图像输入所述第一编码子模型,以获取所述第一编码子模型输出的所述高度特征图像对应的高度编码特征向量;将所述宽度特征图像输入所述第二编码子模型,以获取所述第二编码子模型输出的所述宽度特征图像对应的宽度编码特征向量;所述根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容包括:根据所述文本特征图像、所述高度编码特征向量以及所述宽度编码特征向量,通过所述字符识别子模型,获取所述文本图像对应的文本内容。
125.根据本公开的一个或多个实施例,示例4提供了示例3的方法,所述根据所述文本特征图像、所述高度编码特征向量以及所述宽度编码特征向量,通过所述字符识别子模型,获取所述文本图像对应的文本内容包括:循环执行字符识别步骤,直至所述字符识别子模型输出的目标字符为预设终止字符,将新的文本字符作为所述文本图像对应的文本内容;所述字符识别步骤包括:获取所述文本图像中已识别的文本字符对应的字符嵌入特征向量;将所述文本特征图像、所述高度编码特征向量、所述宽度编码特征向量以及所述字符嵌入特征向量输入所述字符识别子模型,以获取所述字符识别子模型输出的所述目标字符;将所述文本字符和所述目标字符作为新的文本字符。
126.根据本公开的一个或多个实施例,示例5提供了示例4的方法,所述字符识别子模型包括第一解码子模型和第二解码子模型;所述将所述文本特征图像、所述高度编码特征向量、所述宽度编码特征向量以及所述字符嵌入特征向量输入所述字符识别子模型,以获取所述字符识别子模型输出的所述目标字符包括:将所述字符嵌入特征向量输入所述第一解码子模型,以获取所述第一解码子模型输出的所述文本字符对应的字符解码特征向量;将所述文本特征图像、所述高度编码特征向量、所述宽度编码特征向量以及所述字符解码特征向量输入所述第二解码子模型,以获取所述第二解码子模型输出的所述目标字符。
127.根据本公开的一个或多个实施例,示例6提供了示例5的方法,所述第二解码子模型包括多头注意力层和前馈神经网络层,所述多头注意力层包括第一线性子层、缩放点乘积注意力子层、连接子层以及第二线性子层,所述第一线性子层的输出端与所述缩放点乘积注意力子层的输入端耦合,所述缩放点乘积注意力子层的输出端与所述连接子层的输入
端耦合,所述连接子层的输出端与所述第二线性子层的输入端耦合;所述将所述文本特征图像、所述高度编码特征向量、所述宽度编码特征向量以及所述字符解码特征向量输入所述第二解码子模型,以获取所述第二解码子模型输出的所述目标字符包括:将所述高度编码特征向量、所述宽度编码特征向量以及所述字符解码特征向量输入所述第一线性子层,以获取所述第一线性子层输出的目标高度编码特征向量、目标宽度编码特征向量以及目标字符解码特征向量;将所述文本特征图像、所述目标高度编码特征向量、所述目标宽度编码特征向量以及所述目标字符解码特征向量输入所述缩放点乘积注意力子层,以通过所述缩放点乘积注意力子层根据所述文本特征图像、所述目标高度编码特征向量、所述目标宽度编码特征向量以及所述目标字符解码特征向量,确定文本解码特征向量;将所述文本解码特征向量输入所述连接子层,以通过所述连接子层和所述第二线性子层输出所述目标字符。
128.根据本公开的一个或多个实施例,示例7提供了示例6的方法,所述根据所述文本特征图像、所述目标高度编码特征向量、所述目标宽度编码特征向量以及所述目标字符解码特征向量,确定文本解码特征向量包括:根据所述目标高度编码特征向量和所述目标字符解码特征向量,确定高度解码特征向量;根据所述目标宽度编码特征向量和所述目标字符解码特征向量,确定宽度解码特征向量;根据所述高度解码特征向量和所述宽度解码特征向量,确定待定解码特征向量;根据所述待定解码特征向量和所述文本特征图像,确定所述文本解码特征向量。
129.根据本公开的一个或多个实施例,示例8提供了一种文本识别装置,所述装置包括:第一特征图像获取模块,用于获取待识别的文本图像对应的文本特征图像;第二特征图像获取模块,用于根据所述文本特征图像,确定所述文本图像对应的高度特征图像和宽度特征图像,所述高度特征图像用于表征所述文本图像的高度方向的特征,所述宽度特征图像用于表征所述文本图像的宽度方向的特征;文本内容获取模块,用于根据所述文本特征图像、所述高度特征图像以及所述宽度特征图像,通过预先训练的文本识别模型,获取所述文本图像对应的文本内容。
130.根据本公开的一个或多个实施例,示例9提供了一种计算机可读介质,其上存储有计算机程序,该程序被处理装置执行时实现示例1至示例7中任一示例所述方法的步骤。
131.根据本公开的一个或多个实施例,示例10提供了一种电子设备,包括:存储装置,其上存储有计算机程序;处理装置,用于执行所述存储装置中的所述计算机程序,以实现示例1至示例7中任一示例所述方法的步骤。
132.以上描述仅为本公开的较佳实施例以及对所运用技术原理的说明。本领域技术人员应当理解,本公开中所涉及的公开范围,并不限于上述技术特征的特定组合而成的技术方案,同时也应涵盖在不脱离上述公开构思的情况下,由上述技术特征或其等同特征进行任意组合而形成的其它技术方案。例如上述特征与本公开中公开的(但不限于)具有类似功能的技术特征进行互相替换而形成的技术方案。
133.此外,虽然采用特定次序描绘了各操作,但是这不应当理解为要求这些操作以所示出的特定次序或以顺序次序执行来执行。在一定环境下,多任务和并行处理可能是有利的。同样地,虽然在上面论述中包含了若干具体实现细节,但是这些不应当被解释为对本公开的范围的限制。在单独的实施例的上下文中描述的某些特征还可以组合地实现在单个实
施例中。相反地,在单个实施例的上下文中描述的各种特征也可以单独地或以任何合适的子组合的方式实现在多个实施例中。
134.尽管已经采用特定于结构特征和/或方法逻辑动作的语言描述了本主题,但是应当理解所附权利要求书中所限定的主题未必局限于上面描述的特定特征或动作。相反,上面所描述的特定特征和动作仅仅是实现权利要求书的示例形式。关于上述实施例中的装置,其中各个模块执行操作的具体方式已经在有关该方法的实施例中进行了详细描述,此处将不做详细阐述说明。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献