本申请涉及图像处理,具体涉及一种图像风格转换方法、装置、电子设备及存储介质。
背景技术:
1、随着图像处理技术的进步,图像的不同玩法也层出不穷。目前一个比较热门的玩法是对图像的风格进行转换,即将原图转换为一种用户想要的风格的图像。例如,用户将自己的自拍照转换为一张具有漫画风格的自拍照。
2、目前对图像进行风格转换时,先对原图进行编码,然后为编码后的原图添加随机噪声,最后对添加随机噪声后的图像进行去噪,生成风格转换后的图像。
3、然而,目前所生成的风格转换后的图像会丢失原图中的细节信息,导致原图和风格转换后的图像一致性较低,用户体验差。
技术实现思路
1、本申请实施例提供了一种图像风格转换方法、装置、电子设备及存储介质,在对图像进行风格转换时,主动添加原始图像的显性特征,保证原始图像和风格转换后的图像的一致性,提高用户体验。
2、第一方面,本申请实施例提供一种图像风格转换方法,所述方法包括:
3、获取待处理图像;
4、对所述待处理图像进行编码,得到第一特征图;
5、对所述待处理图像进行特征提取,得到至少一个第二特征图,其中,每个第二特征图用于表征所述待处理图像中的一个显性特征;
6、将所述第一特征图与所述至少一个第二特征图进行融合,以对所述待处理图像进行风格转换,得到目标图像。
7、第二方面,本申请实施例提供一种图像风格转换装置,包括:获取单元和处理单元;
8、所述获取单元,用于获取待处理图像;
9、所述处理单元,用于对所述待处理图像进行编码,得到第一特征图;
10、对所述待处理图像进行特征提取,得到至少一个第二特征图,其中,每个第二特征图用于表征所述待处理图像中的一个显性特征;
11、将所述第一特征图与所述至少一个第二特征图进行融合,以对所述待处理图像进行风格转换,得到目标图像。
12、第三方面,本申请实施例提供一种电子设备,包括:处理器和存储器,所述处理器与存储器相连,所述存储器用于存储计算机程序,所述处理器用于执行所述存储器中存储的计算机程序,以使得所述电子设备执行如第一方面所述的方法。
13、第四方面,本申请实施例提供一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序使得计算机执行如第一方面所述的方法。
14、第五方面,本申请实施例提供一种计算机程序产品,所述计算机程序产品包括存储了计算机程序的非瞬时性计算机可读存储介质,所述计算机可操作来使计算机执行如第一方面所述的方法。
15、实施本申请实施例,具有如下有益效果:
16、可以看出,在本申请实施例中,在对待处理图像进行风格转换时,先对图像进行编码,得到第一特征图。然后,对待处理图像进行显性特征的提取,得到至少一个第二特征图,其中,每个第二特征图用于表征待处理图像的一个显性特征。最后,主动将至少一个第二特征图与第一特征图进行融合,这样在对待处理图像进行风格转换时,融合了待处理图像的至少一个显性特征,基于显性特征进行风格转换,可以使转换出的目标图像包含有该至少一个显性特征,则目标图像中包含了待处理图像的细节信息,从而保证了原图和风格转换后的一致性,提高了用户进行图像风格转换的体验。
1.一种图像风格转换方法,其特性在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述将所述第一特征图与所述至少一个第二特征图进行融合,以对所述待处理图像进行风格转换,得到目标图像,包括:
3.根据权利要求2所述的方法,其特征在于,所述将所述第三特征图与每个第二特征图在不同尺度下的多个第四特征图进行融合,以对所述待处理图像进行风格转换,得到目标图像,包括:
4.根据权利要求3所述的方法,其特征在于,所述基于所述多个第五特征图进行多次上采样,并将每个第二特征图在不同尺度下的多个第四特征图分别融合到所述多次上采样过程,得到目标特征图,包括:
5.根据权利要求1-4中任一项所述的方法,其特征在于,
6.根据权利要求2-5中任一项所述的方法,其特征在于,所述第三特征图与每个第二特征图在不同尺度下的多个第四特征图的融合是通过预先训练好的扩散模型实现的;所述扩散模型通过以下步骤训练得到:
7.根据权利要求6所述的方法,其特征在于,所述将所述描述文本输入到初始模型,得到预测图像,包括:
8.一种图像风格转换装置,其特征在于,包括:获取单元和处理单元;
9.一种电子设备,其特征在于,包括:处理器和存储器,所述处理器与所述存储器相连,所述存储器用于存储计算机程序,所述处理器用于执行所述存储器中存储的计算机程序,以使得所述电子设备执行如权利要求1-7中任一项所述的方法。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行以实现如权利要求1-7中任一项所述的方法。