[发明专利]模型训练方法、图像生成方法及相关装置

权利要求书

1.一种模型训练方法，其特征在于，包括：

获取训练样本，所述训练样本包括多组图像，每组图像包括无损图像及其对应的破损图像，所述破损图像由预设的掩码图像对所对应的无损图像处理得到；

将所述多组图像输入至生成模型中，以对所述生成模型进行训练；其中，所述生成模型用于基于破损图像生成所对应的无损图像，所述生成模型基于深度卷积生成对抗网络DCGAN训练得到，所述生成模型包括注意力模块，所述注意力模块用于对接收到的第一特征图进行卷积操作，得到第二特征图，所述第二特征图经所述掩码图像的处理后，用于对所述第一特征图进行加权操作，得到第三特征图，所述注意力模块还用于对所述第三特征图进行卷积操作，得到第四特征图。

2.根据权利要求1所述的方法，其特征在于，所述第一特征图包括所述破损图像中包含的多个像素点的像素值。

3.根据权利要求1所述的方法，其特征在于，所述生成模型还包括卷积层，所述卷积层用于对输入的图像进行卷积操作，得到所述第一特征图。

4.根据权利要求1至3中任一项所述的方法，其特征在于，所述生成模型还包括归一化模块，所述归一化模块用于：

对所述第四特征图和所述掩码图像分别进行归一化操作并求和，得到求和后的数据；

对所述第四特征图进行卷积操作，得到第五特征图；

使用预定义的激活函数对所述第五特征图与所述求和后的数据的乘积进行处理。

5.根据权利要求4所述的方法，其特征在于，所述归一化操作满足如下公式：

其中，y为归一化后的像素值，x为归一化前的像素值，μ为所述归一化前的像素值的均值，S为所述归一化前的像素值的标准差。

6.根据权利要求4所述的方法，其特征在于，所述生成模型包括多个注意力模块和多个归一化模块，所述多个注意力模块与所述多个归一化模块一一对应，每个归一化模块用于从所对应的注意力模块接收所述第四特征图。

7.根据权利要求1所述的方法，其特征在于，所述生成模型还包括反卷积层，用于对接收到的特征图进行反卷积操作。

8.根据权利要求1所述的方法，其特征在于，所述掩码图像包括第一值和第二值，所述第一值所在的位置与所述破损图像中未损坏区域的各像素点对应，所述第二值所在的位置与所述破损图像中损坏区域的各像素点对应，第一值为1，所述第二值为接近0的值。

9.一种图像生成方法，其特征在于，包括：

获取破损图像；

将所述破损图像输入至预训练好的生成模型中，以通过所述生成模型生成所述破损图像的无损图像，所述生成模型基于深度卷积生成式对抗网络DCGAN训练得到，所述生成模型包括注意力模块，所述注意力模块用于对接收到的第一特征图进行卷积操作，得到第二特征图，所述第二特征图经掩码图像的处理后，用于对所述第一特征图进行加权操作，得到第三特征图，所述注意力模块还用于对所述第三特征图进行卷积操作，得到第四特征图；其中，所述第一特征图包括所述破损图像中多个像素点的像素值，或，所述第一特征图为所述破损图像经卷积操作得到的特征图。

10.一种模型训练装置，其特征在于，包括：

获取模块，用于获取训练样本，所述训练样本包括多组图像，每组图像包括无损图像及其对应的破损图像，所述破损图像由预设的掩码图像对所对应的无损图像处理得到；

处理模块，用于将所述多组图像输入至生成模型中，以对所述生成模型进行训练；其中，所述生成模型用于基于破损图像生成所对应的无损图像，所述生成模型基于深度卷积生成式对抗网络DCGAN训练得到，所述生成模型包括注意力模块，所述注意力模块用于对接收到的第一特征图进行卷积操作，得到第二特征图，所述第二特征图经所述掩码图像的处理后，用于对所述第一特征图进行加权操作，得到第三特征图，所述注意力模块还用于对所述第三特征图进行卷积操作，得到第四特征图。