[发明专利]图像转换网络训练方法、装置、计算机设备及存储介质

权利要求书

1.一种图像转换网络训练方法，其特征在于，应用于计算机设备，所述计算机设备中预先存储有参考图像，所述图像转换网络包括第一网络、第二网络及判别网络，所述方法包括以下步骤：

S120、将与参考图像对应的第一图像的下采样图像输入到所述第一网络中进行训练，将所述第一网络输出的图像作为第二图像；

S140、将第一图像输入所述第二网络中进行训练，得到与所述第二图像的上采样图像的内容差值小于第二阈值的第三图像；

S150、将所述第三图像及所述参考图像输入到所述判别网络中；

S160、判别所述第三图像与所述参考图像的特征信息差值是否低于第一阈值；

S170、在所述特征信息差值不低于第一阈值时，将所述第三图像与所述参考图像的特征信息差值反馈给所述第二网络，以对所述第二网络的网络参数进行调整，其中，所述特征信息包括图像边缘特征信息、图像纹理特征信息或图像清晰程度信息；

再次将第一图像输入到调整后的第二网络中，重复上述步骤S140至S170，直到所述特征信息差值小于第一阈值，判定图像转换网络训练完成。

2.根据权利要求1所述的图像转换网络训练方法，其特征在于，所述将所述第三图像与所述参考图像的特征信息差值反馈给所述第二网络，以对所述第二网络的网络参数进行调整，包括：

根据接收到的所述第三图像与所述参考图像的特征信息差值对第二网络的网络参数进行调整，以提升第二网络输出的第三图像的清晰度。

3.根据权利要求1所述的图像转换网络训练方法，其特征在于，在将与参考图像对应的第一图像的下采样图像输入到所述第一网络中进行训练，将所述第一网络输出的图像作为第二图像之前，所述方法还包括：

对所述第一图像进行下采样操作，以得到所述第一图像的下采样图像。

4.根据权利要求1所述的图像转换网络训练方法，其特征在于，在将第一图像输入所述第二网络中进行训练，得到与所述第二图像的上采样图像的内容差值小于第二阈值的第三图像之前，所述方法还包括：

对所述第二图像进行上采样操作，以得到所述第二图像的上采样图像。

5.一种图像转换网络训练装置，其特征在于，应用于计算机设备，所述计算机设备中预先存储有设定图像分辨率的参考图像，所述图像转换网络包括第一网络、第二网络及判别网络，所述装置包括：

第一训练模块，用于将与参考图像对应的第一图像的下采样图像输入到所述第一网络中进行训练，将所述第一网络输出的图像作为第二图像；

第二训练模块，用于将第一图像输入所述第二网络中进行训练，得到与所述第二图像的上采样图像的内容差值小于第二阈值的第三图像；

特征信息判别模块，用于将所述第三图像及所述参考图像输入到所述判别网络中，判别所述第三图像与所述参考图像的特征信息差值是否低于第一阈值，在所述特征信息差值不低于第一阈值时，将所述第三图像与所述参考图像的特征信息差值反馈给所述第二网络，以对所述第二网络的网络参数进行调整；

第二训练模块，还用于再次将第一图像输入到调整后的第二网络中，重复所述第二训练模块执行将第一图像输入所述第二网络中进行训练，得到与所述第二图像的上采样图像的内容差值小于第二阈值的第三图像的步骤和所述特征信息判别模块执行的步骤，直到特征信息差值小于第一阈值，判定图像转换网络训练完成。

6.根据权利要求5所述的图像转换网络训练装置，其特征在于，所述特征信息判别模块包括：

调整子模块，用于根据接收到的所述第三图像与所述参考图像的特征信息差值对第二网络的网络参数进行调整，以提升第二网络输出的图像的清晰度。

7.根据权利要求5所述的图像转换网络训练装置，其特征在于，所述装置还包括：

下采样模块，用于对所述第一图像进行下采样操作，以得到所述第一图像的下采样图像。

8.根据权利要求5所述的图像转换网络训练装置，其特征在于，所述装置还包括：

上采样模块，用于对所述第二图像进行上采样操作，以得到所述第二图像的上采样图像。