[发明专利]一种模型训练及字库建立方法、装置、设备和存储介质

权利要求书

1.一种字体生成模型的训练方法，包括：

将源域样本字输入到字体生成模型的字体生成网络的内容编码器网络，得到样本字内容特征；其中，所述字体生成网络包括：内容编码器网络、随机风格编码器网络、目标风格编码器网络和解码器网络；

将所述源域样本字的目标域关联字输入到所述目标风格编码器网络，得到目标域风格特征；

将所述目标域风格特征和所述样本字内容特征输入到所述解码器网络，得到目标域生成字；

将目标域样本字和所述目标域关联字中的至少一项，以及所述目标域生成字输入到所述字体生成模型的损失分析网络，得到第一损失，并根据所述第一损失调整所述字体生成模型的参数；

将所述源域样本字输入到所述内容编码器网络，得到样本字内容特征；

将随机向量输入到所述随机风格编码器网络，得到随机域风格特征；

将所述随机域风格特征和所述样本字内容特征输入到所述解码器网络，得到随机域生成字；

将所述随机域生成字和随机域样本字输入到所述损失分析网络，得到第二损失，并根据所述第二损失再次调整所述字体生成模型的参数；

其中，所述损失分析网络用于在模型训练阶段，计算训练过程中模型的损失函数。

2.根据权利要求1所述的方法，其中，所述损失分析网络包括：组件分类器网络、字符判别器网络、风格判别器网络和字符分类器网络中的至少一个；所述第一损失包括：组件分类损失、第一字符分类损失、风格分类损失和第一错字损失中的至少一个；所述第二损失包括：第二字符分类损失和/或第二错字损失。

3.根据权利要求2所述的方法，其中，所述将目标域样本字和所述目标域关联字中的至少一项，以及所述目标域生成字输入到所述字体生成模型的损失分析网络，得到第一损失，并根据所述第一损失调整所述字体生成模型的参数，包括：

将目标域样本字和所述目标域生成字输入到所述字符判别器网络，得到第一字符分类损失的第一字符损失值，并基于所述第一字符损失值，调整所述字符判别器网络的参数；

将所述目标域生成字输入到调整后的字符判别器网络，得到第一字符分类损失的第二字符损失值，并基于所述第二字符损失值，调整所述字体生成网络的参数；

将目标域样本字和基于调整后的字体生成网络更新的目标域生成字输入到调整后的字符判别器网络，更新所述第一字符损失值，并基于更新后的第一字符损失值，再次调整字符判别器网络的参数。

4.根据权利要求3所述的方法，其中，所述将目标域样本字和所述目标域生成字输入到所述字符判别器网络，得到第一字符分类损失的第一字符损失值，包括：

将目标域样本字和所述目标域生成字输入到所述字符判别器网络，得到目标域生成字的字符分类头位置和目标域样本字的字符分类头位置；

根据目标域生成字的字符分类头位置和目标域样本字的字符分类头位置，确定第一字符分类损失的第一字符损失值。

5.根据权利要求3所述的方法，其中，所述将所述目标域生成字输入到调整后的字符判别器网络，得到第一字符分类损失的第二字符损失值，包括：

将所述目标域生成字输入到调整后的字符判别器网络，更新目标域生成字的字符分类头位置；

根据更新后的字符分类头位置，确定第一字符分类损失的第二字符损失值。

6.根据权利要求2所述的方法，其中，所述将目标域样本字和所述目标域关联字中的至少一项，以及所述目标域生成字输入到所述字体生成模型的损失分析网络，得到第一损失，并根据所述第一损失调整所述字体生成模型的参数，包括：

将目标域样本字和所述目标域生成字输入到所述风格判别器网络，得到风格分类损失的第一风格损失值，并基于所述第一风格损失值，调整所述风格判别器网络的参数；

将所述目标域生成字输入到调整后的风格判别器网络，得到风格分类损失的第二风格损失值，并基于所述第二风格损失值，调整所述字体生成网络的参数；

将目标域样本字和基于调整后的字体生成网络更新的目标域生成字输入到调整后的风格判别器网络，更新所述第一风格损失值，并基于更新后的第一风格损失值，再次调整风格判别器网络的参数。