[发明专利]图像的转换方法、装置、存储介质和电子装置

说明书

本发明公开了一种图像的转换方法、装置、存储介质和电子装置。其中，该方法包括：获取第一图像转换指令，其中，第一图像转换指令用于指示对待转换的图像进行转换；响应第一图像转换指令，采用第一模型对待转换的图像进行转换，得到目标图像，其中，第一模型用于将第一类别的图像转换为第二类别的图像，待转换的图像为第一类别的图像，目标图像为第二类别的图像，第一类别不同于第二类别，第一模型是通过使用第二类别的样本图像对第一原始模型进行训练得到的模型，第一模型包含第二模型的信息，第二模型用于将第二类别的图像转换为第一类别的图像；输出目标图像。本发明解决了转换得到图像与真实图像不符的技术问题。

技术领域

本发明涉及图像处理领域，具体而言，涉及一种图像的转换方法、装置、存储介质和电子装置。

背景技术

现有技术在进行图像转换时，生成对抗模型，即训练两个模型：生成模型G与鉴别模型D。其中生成模型G用来进行异质(不同类别)图像转换，输入图像x，生成模型G生成与训练样本图像y尽可能像的转换图像G(x)；鉴别模型D用来对图像进行真伪鉴别，鉴别成生成模型G生成的转换图像G(x)为伪，鉴别训练样本图像y为真。生成模型G与鉴别模型D的训练过程可以看作为互相博弈的过程，在互相博弈中逐步提升各自的性能。

但是，上述的生成模型G和鉴别模型D在生成的过程会有一些数据损失，导致最后生成的模型进行图像转换时丢失图像的信息，使得转换得到的图像与真实图像不符。

针对上述的问题，目前尚未提出有效的解决方案。

发明内容

本发明实施例提供了一种图像的转换方法、装置、存储介质和电子装置，以至少解决转换得到图像与真实图像不符的技术问题。

根据本发明实施例的一个方面，提供了一种图像的转换方法，包括：获取第一图像转换指令，其中，所述第一图像转换指令用于指示对待转换的图像进行转换；响应所述第一图像转换指令，采用第一模型对待转换的图像进行转换，得到目标图像，其中，所述第一模型用于将第一类别的图像转换为第二类别的图像，所述待转换的图像为所述第一类别的图像，所述目标图像为所述第二类别的图像，所述第一类别不同于所述第二类别，所述第一模型是通过使用第二类别的样本图像对第一原始模型进行训练得到的模型，所述第一模型包含第二模型的信息，所述第二模型用于将所述第二类别的图像转换为所述第一类别的图像；输出所述目标图像。

根据本发明实施例的另一方面，还提供了一种图像的转换装置，包括：获取单元，用于获取第一图像转换指令，其中，所述第一图像转换指令用于指示对待转换的图像进行转换；响应单元，用于响应所述第一图像转换指令，采用第一模型对待转换的图像进行转换，得到目标图像，其中，所述第一模型用于将第一类别的图像转换为第二类别的图像，所述待转换的图像为所述第一类别的图像，所述目标图像为所述第二类别的图像，所述第一类别不同于所述第二类别，所述第一模型是通过使用第二类别的样本图像对第一原始模型进行训练得到的模型，所述第一模型包含第二模型的信息，所述第二模型用于将所述第二类别的图像转换为所述第一类别的图像；输出单元，用于输出所述目标图像。

在本发明实施例中，通过第一模型对待转换图像进行转换，由于第一模型包含有第二模型的信息，使得第一模型转换得到目标图像保留更多的纹理细节，使得转换得到的图像与真实图像更相似，解决了现有技术转换得到的图像与真实图像不相符的技术问题。

附图说明

此处所说明的附图用来提供对本发明的进一步理解，构成本申请的一部分，本发明的示意性实施例及其说明用于解释本发明，并不构成对本发明的不当限定。在附图中：

图1是根据本发明实施例的一种网络架构的示意图；

图2是根据本发明实施例的图像的转换方法的流程图；

图3是根据本发明实施例的训练两个模型的示意图；

图4是根据本发明实施例的训练模型的网络结构示意图；