模型训练方法、文本去噪方法、装置、设备及存储介质与流程

文档序号:23859618发布日期:2021-02-05 16:01阅读:122来源:国知局
模型训练方法、文本去噪方法、装置、设备及存储介质与流程

[0001]
本发明实施例涉及计算机视觉技术领域,尤其涉及一种模型训练方法、文本去噪方法、装置、设备及存储介质。


背景技术:

[0002]
在各种文本识别场景中,经常会存在文字内容被遮挡或涂抹的情况,例如在对银行表单、财务税票等文本进行识别时,文本图像常被盖上公司公章、公司财务章以及个人名章等图印,导致关键字段无法识别。
[0003]
传统的图印去除方法,常采用对图像的三通道即rgb三个颜色通道的变化进行统计,计算图印的颜色范围区间,并针对对应的颜色范围进行去除。然而该方法对图像的光照、颜色和深浅等干扰鲁棒性较差,且在对图印进行去除后会将图印下的字符一并去除,难以达到文本识别的效果。
[0004]
现有技术中对图印进行去除的方法中也有基于深度学习的图像重建方法,其主要使用超分辨率重建模型(super-resolution reconstruction model,srcnn)对待处理图像中的图印进行去除。然而该方法主要通过平均绝对误差或均方误差对样本图与模型生成图间的像素点距离进行计算,无法准确复现图像中的细节信息,导致生成的图像模糊不清;且该方法对于图像中图印的整体识别效果较差,导致图印无法完全去除,影响文本识别清晰度与准确度。


技术实现要素:

[0005]
本发明提供一种模型训练方法、文本去噪方法、装置、设备及存储介质,以通过新的训练方式对文本去噪神经网络模型进行训练,并通过训练好的文本去噪神经网络模型对待处理图像进行处理,以得到更清晰、准确的文本识别结果。
[0006]
第一方面,本发明实施例提供了一种模型训练方法,该方法包括:
[0007]
确定文本去噪神经网络模型中的当前目标训练层,其中,当前目标训练层中包括至少一个神经网络层;
[0008]
将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;
[0009]
若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。
[0010]
第二方面,本发明实施例还提供了一种文本去噪方法,该方法包括:
[0011]
获取待去噪的文本图像;
[0012]
将待去噪的文本图像输入至预设文本去噪神经网络模型中,其中,预设文本去噪神经网络模型采用本发明实施例提供的模型训练方法训练得到;
[0013]
根据预设文本去噪神经网络模型的输出结果,确定去噪后的目标文本图像。
[0014]
第三方面,本发明实施例还提供了一种模型训练装置,该模型训练装置包括:
[0015]
训练层确定模块,用于确定文本去噪神经网络模型中的当前目标训练层,其中,当前目标训练层中包括至少一个神经网络层;
[0016]
模型训练模块,用于将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;
[0017]
返回确定模块,用于若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。
[0018]
第四方面,本发明实施例还提供了一种文本去噪装置,该文本去噪装置包括:
[0019]
文本图像获取模块,用于获取待去噪的文本图像;
[0020]
文本图像输入模块,用于将待去噪的文本图像输入至预设文本去噪神经网络模型中,其中,预设文本去噪神经网络模型采用本发明实施例提供的模型训练方法训练得到;
[0021]
去噪文本确定模块,用于根据预设文本去噪神经网络模型的输出结果,确定去噪后的目标文本图像。
[0022]
第五方面,本发明实施例还提供了一种计算机设备,包括存储装置、处理器及存储在存储装置上并可在处理器上运行的计算机程序,处理器执行计算机程序时实现如本发明任意实施例提供的方法。
[0023]
第六方面,本发明实施例还提供了一种包含计算机可执行指令的存储介质,计算机可执行指令在由计算机处理器执行时用于执行如本发明任意实施例提供的方法。
[0024]
本发明实施例通过确定文本去噪神经网络模型中的当前目标训练层,其中,当前目标训练层中包括至少一个神经网络层;将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。通过采用上述技术方案,通过图像训练样本分别对文本去噪神经网络模型中的不同目标训练层执行训练,在每次训练中均利用得到的第一损失函数对目标训练层中的权重参数进行调整,使得文本去噪神经网络模型中针对浅层特征的神经网络层可被主要训练调整,进而使得训练得到的文本去噪神经网络模型的整体识别效果更好,解决了文本去噪过程中作为噪音的图印等难以完全去除,影响文本整体识别效果的问题,提高了训练好的文本去噪神经网络模型的文本识别结果的清晰度和准确度。
附图说明
[0025]
图1是本发明实施例一中的一种模型训练方法的流程图;
[0026]
图2是本发明实施例二中的一种模型训练方法的流程图;
[0027]
图3是本发明实施例三中的一种文本去噪方法的流程图;
[0028]
图4是本发明实施例四中的一种模型训练装置的结构示意图;
[0029]
图5是本发明实施例五中的一种文本去噪装置的结构示意图;
[0030]
图6是本发明实施例六中的一种计算机设备的结构示意图。
具体实施方式
[0031]
下面结合附图和实施例对本发明作进一步的详细说明。可以理解的是,此处所描述的具体实施例仅仅用于解释本发明,而非对本发明的限定。另外还需要说明的是,为了便于描述,附图中仅示出了与本发明相关的部分而非全部结构。此外,在不冲突的情况下,本发明中的实施例及实施例中的特征可以互相组合。
[0032]
另外还需要说明的是,为了便于描述,附图中仅示出了与本发明相关的部分而非全部内容。在更加详细地讨论示例性实施例之前应当提到的是,一些示例性实施例被描述成作为流程图描绘的处理或方法。虽然流程图将各项操作(或步骤)描述成顺序的处理,但是其中的许多操作可以被并行地、并发地或者同时实施。此外,各项操作的顺序可以被重新安排。当其操作完成时所述处理可以被终止,但是还可以具有未包括在附图中的附加步骤。所述处理可以对应于方法、函数、规程、子例程、子程序等等。
[0033]
实施例一
[0034]
图1为本发明实施例一提供的一种模型训练方法的流程图,本实施例可适用于对用以进行文本图像识别和文本图像去噪的神经网络模型进行训练的情况,该方法可以由模型训练装置来执行,该模型训练装置可以由软件和/或硬件来实现,该模型训练装置可以配置在计算设备上,具体包括如下步骤:
[0035]
s101、确定文本去噪神经网络模型中的当前目标训练层。
[0036]
其中,当前目标训练层中包括文本去噪神经网络模型中的至少一个神经网络层。
[0037]
在本实施例中,文本去噪神经网络模型可理解为一种用以针对输入的文本图像进行处理并生成去除噪声后的图像的神经网络模型。神经网络模型(neural networks,nn)可理解为一种由大量的、简单的处理单元(也可称为神经元)广泛地互相连接而形成的复杂网络系统,其可反映人脑功能的许多基本特征,是一种高度复杂的非线性动力学习系统,简而言之,神经网络模型可理解为一种以神经元为基础的数学模型。
[0038]
一般的,神经网络模型由多个神经网络层构成,对于一个经典的神经网络其必然包括三个层次,即为输入层、隐藏层和输出层。示例性的,基于现有常用的卷积神经网络(convolutional neural networks,cnn),其中常包括卷积层、全连接层和池化层等,上述各层均可称为神经网络模型中的神经网络层,且各类型的神经网络层均可为一个或多个。
[0039]
在本实施例中,当前目标训练层可理解为根据预设规则确定的在本次针对文本去噪神经网络模型的训练中需要进行训练学习的神经网络层,也即在本次对文本去噪神经网络模型的训练中仅对确定出的当前目标训练层中的权重参数进行调整,而不对文本去噪神经网络模型中除当前目标训练层外的神经网络层中的权重参数进行调整,相当于对文本去噪神经网络模型中除当前目标训练层的神经网络层进行锁定。
[0040]
具体的,根据文本去噪神经网络模型的训练情况,确定当前时刻文本去噪神经网络模型中需要进行训练的神经网络层,并将确定出的神经网络层确定为当前目标训练层。根据训练情况的不同,确定出的当前目标训练层可为文本去噪神经网络模型中的第一个神经网络层,也可为根据上一时刻的当前目标训练层确定的,在上一时刻的当前目标训练层上依次进行递加的新的神经网络层的组合,确定出的当前目标训练层可包括卷积层、全连接层等多种神经网络层类型,其中当前目标训练层的类型可以与文本去噪神经网络模型结构中各神经网络层的分布相关。
[0041]
本发明实施例中,通过确定文本去噪神经网络模型中的当前目标训练层,明确了需要进行训练的神经网络层,避免了在进行神经网络模型训练时对其中所有神经网络层同时进行训练,而导致浅层特征和深层特征同时进行学习重点难以突出的问题,进而可通过当前目标训练层的确定,对文本去噪神经网络模型中的浅层特征进行多次训练,提高训练后模型对整体信息生成的准确性。
[0042]
s102、将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件。
[0043]
在本实施例中,图像训练样本可理解为输入文本去噪神经网络模型中用以对其进行训练的,根据真实图像确定的训练对象的集合,可选的,由于本申请中所需训练的为神经网络模型对文本图像进行去噪的能力,此处输入的图像训练样本可为加噪后的真实图像的集合,进一步地,加噪可理解为在原始的真实图像中添加水印或印章等对文本图像中文字进行遮盖或构成文字背景的内容。损失函数(loss function)可理解为用以度量深度学习过程中训练的模型与真实模型之间距离的函数,损失函数可被用于模型的参数估计(parameteric estimation)使训练的模型达到收敛状态,进而减小训练后模型预测值与真实值之间的误差。
[0044]
具体的,将根据真实图像确定的图像训练样本输入文本去噪神经网络模型中,将图像训练样本中的特征与文本去噪神经网络模型生成的生成图像中的特征代入预设的损失函数中,对其特征距离进行计算得到对应的第一损失函数,利用得到的第一损失函数对文本去噪神经网络模型进行反向传播,使得当前目标训练层中的权重参数可根据第一损失函数进行调整,直到满足预设收敛条件时停止当前对文本去噪神经网络模型的训练。
[0045]
可选的,预设收敛条件可包括计算得到的特征距离小于预设特征距离值、模型训练两次迭代之间的权重参数变化小于预设参数变化阈值、迭代超过设定的最大迭代次数以及训练样本全部训练完毕等,本发明实施例对此不进行限定。
[0046]
s103、若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。
[0047]
具体的,若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,则可认为在当前次对文本去噪神经网络模型的训练过程中,未对文本去噪神经网络模型中的所有神经网络层进行训练,该文本去噪神经网络模型还未训练完成,故此时返回执行步骤s101重新进行当前目标训练层的确定,并再次对文本去噪神经网络模型进行训练,直到文本去噪神经网络模型中的所有神经网络层均进行训练后截止。
[0048]
本实施例的技术方案,通过确定文本去噪神经网络模型中的当前目标训练层,其中,当前目标训练层中包括文本去噪神经网络模型中的至少一个神经网络层;将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;若当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。通过采用上述技术方案,通过同样的图像训练样本分别对文本去噪神经网络模型中的不同目标训练层执行训练,在每次训练中均利用得到的第一损失函数对目标训练层中的权重参数
进行调整,使得文本去噪神经网络模型中针对浅层特征的神经网络层可被主要训练调整,进而使得训练得到的文本去噪神经网络模型的整体识别效果更好,解决了文本去噪过程中作为噪音的图印等难以完全去除,影响文本整体识别效果的问题,提高了训练好的文本去噪神经网络模型的文本识别结果的清晰度和准确度。
[0049]
实施例二
[0050]
图2是本发明实施例二提供的一种模型训练方法的流程图,本发明实施例的技术方案在上述各可选技术方案的基础上进一步优化,通过对具有同一属性特征的图像训练样本中的特征图像与经文本去噪神经网络模型生成图像加噪后的特征图像,求取其特征向量距离进而计算得到与该属性特征对应的第一损失函数,通过第一损失函数对文本去噪神经网络模型中的当前目标训练层进行训练,使得训练后的文本去噪神经网络模型对待处理图像进行处理时可得到整体性更强、真实程度更相近的识别结果。具体包括如下步骤:
[0051]
s201、获取文本去噪神经网络模型的训练情况。
[0052]
可选的,文本去噪神经网络模型可为生成式对抗网络(generative adversarial networks,gan)模型,也可为其他深度学习模型,本发明实施例对此不进行限制。
[0053]
s202、根据文本去噪神经网络模型的训练情况判断文本去噪神经网络模型是否为首次训练,若是,则执行步骤s203;若否,则执行步骤s204。
[0054]
具体的,当文本去噪神经网络模型为首次训练时,可认为当前目标训练层为第一次确定,此时执行步骤s203;当文本去噪神经网络模型不是首次训练时,可认为当前目标训练层不是第一次确定,在上一训练周期中存在对应的当前目标训练层,此时执行步骤s204。
[0055]
s203、将文本去噪神经网络模型中的第一个神经网络层确定为当前目标训练层,并执行步骤s205。
[0056]
具体的,当确定当前目标训练层为首次确定,可认为文本去噪神经网络模型尚未进行训练,此时根据文本去噪神经网络模型对浅层特征多次训练的训练需求,将文本去噪神经网络模型中的第一个神经网络层确定为当前目标训练层,以将除第一个神经网络层外的其他神经网络层进行锁定,在本次训练中仅针对文本去噪神经网络模型中的第一个神经网络层进行训练。
[0057]
s204、在上一当前目标训练层的基础上依次序增加一个文本去噪神经网络模型中的神经网络层,并将增加一个神经网络层后的上一当前目标训练层确定为新的当前目标训练层,并执行步骤s205。
[0058]
具体的,当确定当前目标训练层不是首次确定时,可认为文本去噪神经网络模型已经进行训练但仍未对其中的神经网络层完全进行训练,此时依据文本去噪神经网络模型中各神经网络层的排布顺序,在上一训练周期的上一当前目标训练层的基础上依照排布顺序增加一个神经网络层,并将增加一个神经网络层后的上一当前目标训练层确定为新的当前目标训练层,以对文本去噪神经网络模型进行神经网络层逐次递增的训练。
[0059]
在本发明实施例中,通过当前目标训练层的确定使得文本去噪神经网络模型中排于前部的神经网络层得到更多次数的训练,由于模型中排于前部的神经网络层主要用以学习浅部特征,而浅部特征主要对应输入图像中的整体信息,从而使得通过该方法训练的文本去噪神经网络模型针对待处理图像中的整体信息获取效果更强,得到的文本识别结果更清晰。
[0060]
s205、将图像训练样本输入至文本去噪神经网络模型,得到文本去噪神经网络模型输出的生成图像。
[0061]
具体的,将根据真实图像确定的图像训练样本输入至文本去噪神经网络模型中,文本去噪神经网络模型根据其中各神经网络层的权重参数及对应的属性参数输出生成图像,该生成图像可理解为与输入的图像训练样本相对应的,未添加噪声的图像,也即是希望生成的与真实图像相近的生成图像。
[0062]
进一步地,输入文本去噪神经网络模型的图像训练样本可理解为对真实图像进行噪声添加后所生成的加噪图像的集合,本发明实施例还提供了一种图像训练样本的确定方法,具体包括如下步骤:
[0063]
s2051、获取真实图像样本。
[0064]
具体的,获取未进行加噪的文本图像,加噪可理解为在文本图像上添加印章或水印等使其与文本图像文字产生交叠,影响文本图像文字显示清晰度及完整度的操作,真实图像样本可理解为原始的文本图像,也即文本去噪神经网络模型希望其输出所接近的图像。
[0065]
s2052、根据预设噪声添加函数对真实图像样本进行噪声添加。
[0066]
在本实施例中,预设噪声添加函数可理解为一种用以模拟真实环境下对文本图像进行印章或水印添加的函数,其通过对输入的真是样本图像中预设位置像素点信息进行修改,使得真实图像样本中原有文本信息被遮蔽,即可实现对真实图像样本的噪声添加,也即输入文本去噪神经网络模型中的图像训练样本与真实图像样本间存在映射关系。
[0067]
s2053、将添加噪声后的真实图像样本确定为图像训练样本。
[0068]
s206、根据预设噪声添加函数对生成图像进行噪声添加。
[0069]
具体的,根据与生成图像训练样本相同的预设噪声添加函数对通过文本去噪神经网络模型生成的生成图像进行噪声添加,使得添加噪声后的生成图像与输入文本去噪神经网络模型的图像训练样本状态类似,进而可通过比较加噪后生成图像与图像训练样本的相似度,确定生成图像与未加噪的图像训练样本间的相似度。由于二者添加噪声的方式相同,故当加噪后的生成图像与图像训练样本间的相似度高时,生成图像与原始真实图像间的相似度也高,进而可说明训练后的文本去噪神经网络模型在对待处理的图像进行处理后所生成的图像更贴近于没有噪声的真实图像。
[0070]
s207、将添加噪声后的生成图像与图像训练样本进行比对,根据比对结果确定文本去噪神经网络模型对应的第一损失函数。
[0071]
具体的,将添加噪声后的生成图像与对应的图像训练样本分别划分为多张特征图,每个特征图中可以仅包含该图像中的一种属性特征,确定出添加噪声后的生成图像与对应的图像训练样本中对应同一种属性特征的两张特征图间的特征距离,根据得到的特征距离综合确定比对结果,根据上述比对结果确定出文本去噪神经网络模型对应的第一损失函数。
[0072]
其中,属性特征可理解为光照、颜色、深浅、图像中文本整体形状等用以表征图像的整体显示状态的特征。
[0073]
进一步地,本发明实施例还提供了一种根据比对结果确定文本去噪神经网络模型对应的第一损失函数的方法,具体包括如下步骤:
[0074]
s2071、确定与所述添加噪声后的生成图像的第一特征图像相对应的所述图像训练样本的第二特征图像;
[0075]
其中,所述第一特征图像与所述第二特征图像具有相同的属性特征。
[0076]
s2072、确定第一特征图像与第二特征图像间的特征向量距离。
[0077]
s2073、根据特征向量距离确定文本去噪神经网络模型对应的第一损失函数。
[0078]
在本实施例中,特征向量距离可理解为用以衡量第一特征图像与第二特征图像在视觉上的接近程度的参量。
[0079]
具体的,将添加噪声后的生成图像中具有某一种属性特征的特整图像确定为第一特征图像,将输入文本去噪神经网络模型用以生成该添加噪声后的生成图像的图像训练样本中,具有同上述第一特征图像具备同样属性特征的特征图确定为第二特征图像。基于不同的属性特征的特性通过神经网络模型中的鉴别其确定第一特征图像与第二特征图像之间的特征向量距离,进而确定文本去噪神经网络模型在本次训练过程中对应的第一损失函数。
[0080]
示例性的,在本发明实施例中提供了一种第一损失函数的确定方法,该方法通过如下表达式实现:
[0081][0082]
其中,y表示对真实图像加噪后生成的图像训练样本,m表示加噪后的生成图像,d表示生成式对抗网络中的鉴别器,d(y,i)表示y在鉴别器中的第i个特征图,也即图像训练样本的第二特征图像,d(m,i)表示m在鉴别器中的第i个特征图,也即加噪后的生成图像对应的第一特征图像,l(y,m)表示根据加噪后的生成图像以及图像训练样本的各属性特征确定的第一损失函数。
[0083]
在上述公式中,y可用如下公式表示:
[0084]
y=φ(x)
[0085]
其中,x表示现实中的真实图像样本,φ表示用以加盖印章等噪声的加噪函数,y表示对真实图像加噪后生成的图像训练样本。
[0086]
在上述公式中,m可用如下公式表示:
[0087]
m=φ(g(z;θ))
[0088]
其中,g表示生成式对抗网络中的生成器,θ表示生成器中的参数,z表示先验特征向量。
[0089]
s208、基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件。
[0090]
s209、判断训练后的当前目标训练层中是否包括文本去噪神经网络模型中的所有神经网络层,若是,则执行步骤s210;若否,则执行步骤s201。
[0091]
具体的,根据当前目标训练层中的神经网络层层数确定当前目标训练层中是否已经包括文本去噪神经网络模型中的所有神经网络层,也即是否在本次训练中对文本去噪神经网络模型进行完整的训练,若是,则认为该文本去噪神经网络模型已训练完成,进而执行步骤s210;若否,则认为该文本去噪神经网络模型尚未训练完成,需重新对当前目标训练层进行确定,进而返回执行步骤s201。
[0092]
s210、将文本去噪神经网络模型确定为训练好的文本去噪神经网络模型。
[0093]
示例性的,假设待训练的文本去噪神经网络模型中具有5个卷积层,则在训练过程中首先判断当前目标训练层的确定次数,也即判断文本去噪神经网络模型是否是第一次进行训练,当确定其是第一次进行训练时,将当前目标训练层确定为上述5个卷积层中的第1层,在当次训练中固定第2-5层的参数,仅通过输入的图像训练样本对第1层的参数进行学习。进一步地,在第二次进行训练时,将当前目标训练层确定为上述5个卷积层中的第1-2层,在当次训练中固定第3-5层的参数,仅通过输入的图像训练样本对第1-2层的参数进行学习,以此类推,直至5层参数在一次训练中全部进行学习,则认为在该次学习后文本去噪神经网络模型已被训练好。
[0094]
本实施例的技术方案,通过对真实图像添加噪声得到输入文本去噪神经网络模型的图像训练样本,根据文本去噪神经网络模型的训练情况确定当前目标训练层,根据具有同一属性特征的图像训练样本中的特征图像与经文本去噪神经网络模型生成图像加噪后的特征图像确定其间的特征向量距离,进而通过特征向量距离确定用以对模型进行训练的第一损失函数,利用当次确定的第一损失函数经反向传播对文本去噪神经网络模型中的当前目标训练层进行训练,直到当前目标训练层中包括文本去噪神经网络模型中的所有神经网络层时确定模型训练完成,使得训练完成的文本去噪神经网络模型在对待处理图像进行处理时可得到整体性更强、清晰度更高、对真实图像还原程度更好的识别结果。
[0095]
实施例三
[0096]
图3为本发明实施例三提供的一种文本去噪方法的流程图,该方法可以由文本去噪装置执行,其中该文本去噪装置可以由软件和/或硬件来实现,该文本去噪装置可以配置在计算设备上,具体包括如下步骤:
[0097]
s301、获取待去噪的文本图像。
[0098]
示例性的,待去噪的文本图像可以与本发明实施例中的图像训练样本的形式一致。示例性的,待去噪的文本图像可为通过扫描获得的加盖印章的合同图像等。
[0099]
s302、将待去噪的文本图像输入至预设文本去噪神经网络模型中。
[0100]
其中,所述预设文本去噪神经网络模型采用本发明实施例提供的一种模型训练方法训练得到。
[0101]
s303、根据预设文本去噪神经网络模型的输出结果,确定去噪后的目标文本图像。
[0102]
示例性的,将待去噪的文本图像输入预设文本去噪神经网络模型后,预设文本去噪神经网络模型可根据其内部训练好的参数就输入的待去噪的文本图像进行图像生成,生成的图像即为去噪后的目标文本图像。
[0103]
本发明实施例提供的文本去噪方法,由于采用了本发明实施例提供的模型训练方法训练得到的文本去噪神经网络模型,再基于该模型进行文本去噪,能够对输入的待去噪文本图像中的整体特征起到较好的识别效果,提高了生成的去噪后的目标文本图像的清晰度和准确度,以及其与真实图像的相似度。
[0104]
实施例四
[0105]
图4为本发明实施例四提供的一种模型训练装置的结构示意图,该模型训练装置包括:训练层确定模块41,模型训练模块42和返回确定模块43。
[0106]
其中,训练层确定模块41,用于确定文本去噪神经网络模型中的当前目标训练层,
其中,当前目标训练层中包括至少一个神经网络层;模型训练模块42,用于将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;返回确定模块43,用于若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。
[0107]
本实施例的技术方案,解决了文本去噪过程中作为噪音的图印等难以完全去除,影响文本整体识别效果的问题,提高了训练好的文本去噪神经网络模型的文本识别结果的清晰度和准确度。
[0108]
可选的,模型训练装置还包括:
[0109]
训练样本确定模块,用于获取真实图像样本;根据预设噪声添加函数对真实图像样本进行噪声添加;将添加噪声后的真实图像样本确定为图像训练样本。
[0110]
可选的,训练层确定模块41,具体用于:若文本去噪神经网络模型为首次训练,将所述文本去噪神经网络模型中的第一个神经网络层确定为所述当前目标训练层;若文本去噪神经网络模型不是首次训练,则在上一当前目标训练层的基础上依次序增加一个所述文本去噪神经网络模型中的神经网络层,并将增加一个神经网络层后的上一当前目标训练层确定为新的当前目标训练层。
[0111]
可选的,模型训练模块42,具体用于:将图像训练样本输入至文本去噪神经网络模型,得到文本去噪神经网络模型输出的生成图像;根据预设噪声添加函数对生成图像进行噪声添加;将添加噪声后的生成图像与图像训练样本进行比对,根据比对结果确定文本去噪神经网络模型对应的第一损失函数;基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件。
[0112]
进一步地,将添加噪声后的生成图像与图像训练样本进行比对,根据比对结果确定文本去噪神经网络模型对应的第一损失函数,包括:确定与所述添加噪声后的生成图像的第一特征图像相对应的所述图像训练样本的第二特征图像,其中,所述第一特征图像与所述第二特征图像具有相同的属性特征;确定第一特征图像与第二特征图像间的特征向量距离;根据特征向量距离确定文本去噪神经网络模型对应的第一损失函数。
[0113]
进一步地,若当前目标训练层中已包括文本去噪神经网络模型中的所有神经网络层,则将文本去噪神经网络模型确定为训练好的文本去噪神经网络模型。
[0114]
本发明实施例提供的模型训练装置可执行本发明任意实施例所提供的模型训练方法,具备执行方法相应的功能模块和有益效果。
[0115]
实施例五
[0116]
图5是本发明实施例五提供的一种文本去噪装置的结构示意图,该文本去噪装置可由软件和/或硬件实现,一般可集成在计算机设备中,可通过执行文本去噪方法来进行文本去噪。如图5所示,该装置包括:文本图像获取模块51,文本图像输入模块52和去噪文本确定模块53。
[0117]
其中,文本图像获取模块51,用于获取待去噪的文本图像;文本图像输入模块52,用于将待去噪的文本图像输入至预设文本去噪神经网络模型中,其中,预设文本去噪神经
网络模型采用本发明实施例提供的模型训练方法训练得到;去噪文本确定模块53,用于根据预设文本去噪神经网络模型的输出结果,确定去噪后的目标文本图像。
[0118]
本发明实施例提供的文本去噪装置,由于采用了本发明实施例提供的模型训练方法训练得到的文本去噪神经网络模型,再基于该模型进行文本去噪,能够对输入的待去噪文本图像中的整体特征起到较好的识别效果,提高了生成的去噪后的目标文本图像的清晰度和准确度,以及其与真实图像的相似度。
[0119]
实施例六
[0120]
图6为本发明实施例六提供的一种计算机设备的结构示意图,该计算机设备中可集成本发明实施例提供的模型训练装置和/或文本去噪装置。如图6所示,计算机设备600包括存储装置601、处理器602及存储在存储装置601上并可在处理器602上运行的计算机程序,所述处理器602执行所述计算机程序时实现本发明实施例提供的模型训练方法和/或文本去噪方法。
[0121]
存储装置601作为一种计算机可读存储介质,可用于存储软件程序、计算机可执行程序以及模块,如本发明实施例中的模型训练方法和/或文本去噪方法对应的程序指令/模块(例如,训练层确定模块41,模型训练模块42和返回确定模块43,或,文本图像获取模块51,文本图像输入模块52和去噪文本确定模块53)。处理器602通过运行存储在存储装置601中的软件程序、指令以及模块,从而执行设备的各种功能应用以及数据处理,即实现上述的模型训练方法和/或文本去噪方法。
[0122]
存储装置601可主要包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需的应用程序;存储数据区可存储根据终端的使用所创建的数据等。此外,存储装置601可以包括高速随机存取存储器,还可以包括非易失性存储器,例如至少一个磁盘存储器件、闪存器件、或其他非易失性固态存储器件。在一些实例中,存储装置601可进一步包括相对于处理器602远程设置的存储器,这些远程存储器可以通过网络连接至识别。上述网络的实例包括但不限于互联网、企业内部网、局域网、移动通信网及其组合。
[0123]
实施例七
[0124]
本发明实施例五还提供一种包含计算机可执行指令的存储介质,所述计算机可执行指令在由计算机处理器执行时用于执行本发明实施例提供的模型训练方法和/或文本去噪方法。
[0125]
其中,模型训练方法包括:
[0126]
确定文本去噪神经网络模型中的当前目标训练层,其中,当前目标训练层中包括至少一个神经网络层;
[0127]
将图像训练样本输入至文本去噪神经网络模型,并计算对应的第一损失函数,基于第一损失函数对文本去噪神经网络模型进行训练,以对当前目标训练层中的权重参数进行调整,直到文本去噪神经网络模型满足预设收敛条件;
[0128]
若训练后的当前目标训练层中未包括文本去噪神经网络模型中的所有神经网络层,返回执行确定文本去噪神经网络模型中的当前目标训练层的步骤。
[0129]
其中,文本去噪方法,包括:
[0130]
获取待去噪的文本图像;
[0131]
将待去噪的文本图像输入至预设文本去噪神经网络模型中,其中,预设文本去噪
神经网络模型采用本发明实施例提供的模型训练方法训练得到;
[0132]
根据预设文本去噪神经网络模型的输出结果,确定去噪后的目标文本图像。
[0133]
当然,本发明实施例所提供的一种包含计算机可执行指令的存储介质,其计算机可执行指令不限于如上所述的方法操作,还可以执行本发明任意实施例所提供的模型训练方法和/或文本去噪方法中的相关操作。
[0134]
通过以上关于实施方式的描述,所属领域的技术人员可以清楚地了解到,本发明可借助软件及必需的通用硬件来实现,当然也可以通过硬件实现,但很多情况下前者是更佳的实施方式。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,如计算机的软盘、只读存储器(read-only memory,rom)、随机存取存储器(random access memory,ram)、闪存(flash)、硬盘或光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本发明各个实施例所述的方法。
[0135]
值得注意的是,上述搜索装置的实施例中,所包括的各个单元和模块只是按照功能逻辑进行划分的,但并不局限于上述的划分,只要能够实现相应的功能即可;另外,各功能单元的具体名称也只是为了便于相互区分,并不用于限制本发明的保护范围。
[0136]
注意,上述仅为本发明的较佳实施例及所运用技术原理。本领域技术人员会理解,本发明不限于这里所述的特定实施例,对本领域技术人员来说能够进行各种明显的变化、重新调整和替代而不会脱离本发明的保护范围。因此,虽然通过以上实施例对本发明进行了较为详细的说明,但是本发明不仅仅限于以上实施例,在不脱离本发明构思的情况下,还可以包括更多其他等效实施例,而本发明的范围由所附的权利要求范围决定。
当前第1页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1