[发明专利]字迹修复方法、装置、电子设备和存储介质

说明书

本发明提供一种字迹修复方法、装置、电子设备和存储介质，其中方法包括：获取目标风格下的破损字迹图像；将所述破损字迹图像输入至字迹修复模型，得到所述字迹修复模型输出的修复字迹图像；所述字迹修复模型是在预训练生成模型的基础上，应用所述目标风格下的第一样本字迹图像，结合判别模型进行生成对抗训练得到的；所述预训练生成模型是基于各种风格下的第二样本字迹图像训练得到的，所述判别模型用于区分真实图像和合成图像。本发明提供的方法、装置、电子设备和存储介质，降低了模型训练所需样本的获取难度，提高了模型训练效率，字迹修复图像的自然度和逼真度得以保证，实现了不同风格的书法修复的自动化。

技术领域

本发明涉及人工智能技术领域，尤其涉及一种字迹修复方法、装置、电子设备和存储介质。

背景技术

随着时间的推移，很多流传的书法作品由于保管不慎或者自然老化，存在字迹不清或者局部污损残缺的情况。针对上述问题的修复，需要精通书法艺术的专业人员依据相当多的背景知识和基本技巧来进行手工模拟临摹。

然而，单靠人力进行修复，所需的成本和时间难以估量。因此，如何实现低成本、高效率的字迹修复，依然是亟待解决的问题。

发明内容

本发明提供一种字迹修复方法、装置、电子设备和存储介质，用以解决现有技术中依赖人工进行字迹修复，费时费力的缺陷。

本发明提供一种字迹修复方法，包括：

获取目标风格下的破损字迹图像；

将所述破损字迹图像输入至字迹修复模型，得到所述字迹修复模型输出的修复字迹图像；

所述字迹修复模型是在预训练生成模型的基础上，应用所述目标风格下的第一样本字迹图像，结合判别模型进行生成对抗训练得到的；所述预训练生成模型是基于各种风格下的第二样本字迹图像训练得到的，所述判别模型用于区分真实图像和合成图像。

根据本发明提供的一种字迹修复方法，所述字迹修复模型基于如下步骤训练得到：

对所述第一样本字迹图像进行随机遮盖，将随机遮盖后的第一样本字迹图像作为第一破损图像，将所述第一样本字迹图像中被随机遮盖的部分作为第一剪切图像；

将所述第一破损图像输入至所述预训练生成模型，得到所述预训练生成模型输出的第一修复图像；

将所述第一修复图像和所述第一剪切图像分别输入至所述判别模型，得到所述判别模型输出的所述第一修复图像的判别结果和所述第一剪切图像的判别结果；

基于所述第一修复图像和所述第一剪切图像，以及所述第一修复图像的判别结果和所述第一剪切图像的判别结果，对所述预训练生成模型进行参数迭代，得到所述字迹修复模型。

根据本发明提供的一种字迹修复方法，所述基于所述第一修复图像和所述第一剪切图像，以及所述第一修复图像的判别结果和所述第一剪切图像的判别结果，对所述预训练生成模型进行参数迭代，得到所述字迹修复模型，包括：

基于所述第一修复图像和所述第一剪切图像，确定重构损失；

基于所述第一修复图像的判别结果和所述第一剪切图像的判别结果，确定对抗损失；

基于所述重构损失和所述对抗损失，对所述预训练生成模型进行参数迭代，得到所述字迹修复模型。

根据本发明提供的一种字迹修复方法，所述基于所述重构损失和所述对抗损失，对所述预训练生成模型进行参数迭代，得到所述字迹修复模型，包括：

基于所述重构损失，对所述预训练生成模型进行参数迭代，并且，基于所述对抗损失，对所述预训练生成模型和所述判别模型进行参数迭代，得到所述字迹修复模型。