[发明专利]图像处理模型的训练方法、文档图像处理方法及设备

说明书

本申请提供一种图像处理模型的训练方法、文档图像处理方法及设备。该方法包括：在多张无噪白底黑字图像中分别加入噪声，得到多张有噪白底黑字图像；将有噪白底黑字图像与多张背景图分别进行融合，得到多张样本文档图像，多张背景图为多种不同的场景图像；对无噪白底黑字图像进行反色处理，得到无噪黑底白字图像；将样本文档图像和无噪黑底白字图像进行一一对应，得到训练样本集，训练样本集包括：多组样本图像，每组样本图像包括：一张样本文档图像和对应的无噪黑底白字图像；根据训练样本集进行模型训练，得到图像处理模型，以对输入的文档图像进行去噪和黑白化处理。本申请训练的图像处理模型能更好地保留文字内容，可同时处理多个文档图像。

技术领域

本申请涉及图像处理领域，具体而言，涉及一种图像处理模型的训练方法、文档图像处理方法及设备。

背景技术

随着社会的高速发展，各种高新技术层出不穷。图像作为一种可传递大量信息的载体，在诸多领域被广泛的应用。但是，图像在数字化和传输过程中会受到成像设备和外部环境噪声的影响，导致图像的可视化效果变差，影响到对图像中有效信息的提取。例如，在现实场景中，为获取文档图像，会对文档图像进行电子扫描、拍照等处理，处理时的场景可能会对文档图像有一定的干扰，使得文档图像变得模糊不清，并不能够完整清楚地展示图像中的文字内容，因此，需要对获取到的文档图像进行处理，以清楚地展示文档图像中的文字内容。

但现有的图像去噪方法，仅仅只是对噪声图像进行去噪，使得图像变得清晰一些，并不改变原图像的显示结构，泛化能力弱，无法对来自不同场景的图像进行去噪。在处理文档图像时，会将一个文档图像看做一个整体进行去噪清晰化处理，并不能针对性地处理文档图像中的文字内容，依然无法清楚地展示文档图像中的文字内容。

发明内容

本发明的目的在于，针对上述现有技术中的不足，提供一种图像处理模型的训练方法、文档图像处理方法及设备，以解决现有技术中，无法清楚地展示文档图像中的文字内容等问题。

为实现上述目的，本申请实施例采用的技术方案如下：

第一方面，本申请实施例提供一种图像处理模型的训练方法，该方法包括：

在多张无噪白底黑字图像中分别加入噪声，得到多张有噪白底黑字图像；

将所述有噪白底黑字图像与多张背景图分别进行融合，得到多张样本文档图像，其中，所述多张背景图分别为多种不同的场景图像；

对所述无噪白底黑字图像进行反色处理，得到无噪黑底白字图像；

将所述样本文档图像和所述无噪黑底白字图像进行一一对应，得到训练样本集，所述训练样本集包括：多组样本图像，每组样本图像包括：一张样本文档图像和对应的无噪黑底白字图像；

根据所述训练样本集进行模型训练，得到图像处理模型，以对输入的文档图像进行去噪和黑白化处理。

可选地，所述在多张无噪白底黑字图像中分别加入噪声，得到多张有噪白底黑字图像，包括：

分别采用至少一组噪声参数，在所述无噪的白底黑字图像中加入噪声，得到至少一组所述有噪白底黑字图像，每组所述有噪高清的白底黑字图像对应一组噪声参数。

可选地，所述无噪白底黑字图像为无噪高清的白底黑字图像，相应的，所述有噪白底黑字图像为有噪高清的白底黑字图像；

所述将所述有噪白底黑字图像与多张背景图分别进行融合，得到多张样本文档图像，包括：

对所述有噪高清的白底黑字图像进行滤波，得到有噪模糊的白底黑字图像；

将所述有噪模糊的白底黑字图像与所述多张背景图分别进行融合，得到多张所述样本文档图像。

可选地，所述对所述有噪高清的白底黑字图像进行滤波，得到有噪模糊的白底黑字图像，包括：