[发明专利]图像修复模型的训练方法及图像修复方法、装置及设备

权利要求书

1.一种图像修复模型的训练方法，其特征在于，所述方法包括：

将真实第一像素质量图像输入图像特征编码器，得到第一图像特征；

将人工合成第一像素质量图像输入所述图像特征编码器，得到第二图像特征；所述人工合成第一像素质量图像是由真实第二像素质量图像进行模糊处理得到的；

将所述第一图像特征输入真实第一像素质量图像生成器，得到重新生成的真实第一像素质量图像；

将所述第二图像特征输入所述真实第一像素质量图像生成器，得到伪真实第一像素质量图像；

将所述第二图像特征输入真实第二像素质量图像生成器，得到重建第二像素质量图像；

根据所述第一图像特征以及所述第二图像特征计算域对齐损失；

根据所述重新生成的真实第一像素质量图像、所述伪真实第一像素质量图像以及所述真实第一像素质量图像计算图像生成损失；

根据所述重建第二像素质量图像以及所述真实第二像素质量图像计算图像重建损失；

根据所述域对齐损失、所述图像生成损失以及所述图像重建损失，训练所述图像特征编码器、所述真实第一像素质量图像生成器以及所述真实第二像素质量图像生成器，重复执行所述将真实第一像素质量图像输入图像特征编码器，得到第一图像特征以及后续步骤，直到达到预设条件；

其中，第一像素质量的图像清晰度低于第二像素质量的图像清晰度。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

将所述伪真实第一像素质量图像输入所述图像特征编码器，得到第三图像特征；

根据所述第二图像特征以及所述第三图像特征计算内容一致性损失；

所述根据所述域对齐损失、所述图像生成损失以及所述图像重建损失，训练所述图像特征编码器、所述真实第一像素质量图像生成器以及所述真实第二像素质量图像生成器，包括：

所述根据所述域对齐损失、所述图像生成损失、所述图像重建损失以及所述内容一致性损失，训练所述图像特征编码器、所述真实第一像素质量图像生成器以及所述真实第二像素质量图像生成器。

3.根据权利要求2所述的方法，其特征在于，所述方法还包括：

将所述第三图像特征输入所述真实第二像素质量图像生成器，得到伪真实重建第二像素质量图像；

所述根据所述重建第二像素质量图像以及所述真实第二像素质量图像计算图像重建损失，包括：

根据所述重建第二像素质量图像以及所述真实第二像素质量图像计算第一图像重建损失；

根据所述伪真实重建第二像素质量图像以及所述真实第二像素质量图像计算第二图像重建损失；所述第一图像重建损失以及所述第二图像重建损失组成图像重建损失。

4.根据权利要求1所述的方法，其特征在于，所述根据所述重新生成的真实第一像素质量图像、所述伪真实第一像素质量图像以及所述真实第一像素质量图像计算图像生成损失，包括：

根据所述重新生成的真实第一像素质量图像以及所述真实第一像素质量图像计算第一图像生成损失；

根据所述伪真实第一像素质量图像以及所述真实第一像素质量图像计算第二图像生成损失；所述第一图像生成损失以及所述第二图像生成损失组成图像生成损失。

5.根据权利要求1所述的方法，其特征在于，所述根据所述第一图像特征以及所述第二图像特征计算域对齐损失，包括：

将所述第一图像特征输入特征判别器，得到第一概率值；

将所述第二图像特征输入所述特征判别器，得到第二概率值；

根据所述第一概率值以及所述第二概率值，计算域对齐损失。

6.根据权利要求2所述的方法，其特征在于，所述根据所述第二图像特征以及所述第三图像特征计算内容一致性损失，包括：

计算所述第二图像特征以及所述第三图像特征之间的1-范数；

根据所述第二图像特征以及所述第三图像特征之间的1-范数，计算内容一致性损失。