[发明专利]文档生成器的获取方法、文档生成方法、装置及电子设备

权利要求书

1.一种文档生成器的获取方法，其特征在于，所述文档生成器用于根据拍摄的文档图像生成电子文档图像，所述方法包括：

从第一图像集选取一个图像作为源域图像，从第二图像集选取一个图像作为目标域图像，所述第一图像集包括拍摄的多个文档图像，所述第二图像集包括多个电子文档的图像，定义第一图像集中的图像所在的图像域为源域，定义第二图像集中的图像所在的图像域为目标域；

将所述源域图像以及所述目标域图像输入循环一致性生成对抗网络，生成由所述源域图像映射到所述目标域的映射图像以及由所述映射图像映射到源域的重建图像，并获取所述映射图像与所述目标域图像之间的判别损失以及所述源域图像与所述重建图像之间的重建损失；

获取所述源域图像的边缘图像与所述映射图像之间的边缘损失；

将所述源域图像以及所述目标域图像输入另一循环一致性生成对抗网络，生成由所述目标域图像映射到所述源域的另一映射图像以及由所述另一映射图像映射到目标域的另一重建图像，并获取所述另一映射图像与所述目标域图像之间的判别损失，以及获取所述源域图像与所述另一重建图像之间的重建损失，所述另一循环一致性生成对抗网络的生成器与所述循环一致性生成对抗网络的生成器相同；

获取所述目标域图像的边缘图像与所述另一映射图像之间的边缘损失；

根据所述判别损失、重建损失以及边缘损失的和对所述循环一致性生成对抗网络中的生成器以及判别器进行优化；

所述根据所述判别损失、重建损失以及边缘损失的和对所述循环一致性生成对抗网络中的生成器以及判别器进行优化包括：

根据所有的判别损失、重建损失以及边缘损失的和对所述循环一致性生成对抗网络以及所述另一循环一致性生成对抗网络中的生成器以及判别器进行优化；

循环执行所述从第一图像集选取一个图像作为源域图像，从第二图像集选取一个图像作为目标域图像的步骤，至所述根据所述判别损失、重建损失以及边缘损失的和对所述循环一致性生成对抗网络中的生成器以及判别器进行优化的步骤，直至所述循环一致性生成对抗网络收敛时，以由源域图像生成目标域图像的生成器作为所述文档生成器。

2.根据权利要求1所述的方法，其特征在于，所述根据所述判别损失、重建损失以及边缘损失的和对所述循环一致性生成对抗网络中的生成器以及判别器进行优化，包括：

最大化判别器的损失，最小化生成器的损失，将所述判别损失、重建损失以及边缘损失的和通过反向传播算法对所述循环一致性生成对抗网络中的生成器的参数以及判别器的参数进行优化。

3.根据权利要求1所述的方法，其特征在于，所述获取所述源域图像的边缘图像与所述映射图像之间的边缘损失，包括：

对所述源域图像进行边缘检测，获得所述源域图像的边缘图像；

计算所述源域图像的边缘图像与所述映射图像之间的边缘损失：

L_edge(G)＝E[||G(x)-Ed(X)||₁]，

其中，所述G(X)表示源域图像的映射图像，Ed(X)表示源域图像的边缘图像。

4.根据权利要求1所述的方法，其特征在于，所述从第一图像集选取一个图像作为源域图像，从第二图像集选取一个图像作为目标域图像之前，还包括：

对所述第一图像集以及所述第二图像集中的图像进行增强处理，以增加第一图像集中的数据量以及所述第二图像集中的数据量。

5.根据权利要求1所述的方法，其特征在于，所述从第一图像集选取一个图像作为源域图像，从第二图像集选取一个图像作为目标域图像之前，还包括：

以拍摄的多个纸质文档的图像作为所述第一图像集中的图像；

按照所述纸质文档的文字排列方式，编辑文字大小在所述纸质文档中文字大小的预设范围内的多个电子文档；

获取所述多个电子文档的图像作为所述第二图像集中的图像。

6.一种文档生成方法，其特征在于，用于根据权利要求1-5任一项所述的方法获取的文档生成器生成电子文档图像，所述方法包括：

获取拍摄纸质文档获得的文档图像；

将所述文档图像输入所述文档生成器；

以所述文档生成器生成的图像作为所述文档图像对应的电子文档图像。