本公开涉及图像处理领域,具体涉及一种图像修复方法、装置、计算机设备及存储介质。
背景技术:
1、目前的图像修复方案,虽然已有各领域的分类模型和图像局部修复技术,但是在修复过程中需要人工进行干预。例如:在常见的图像生成技术中,图像局部绘制功能仍然需要用户手动圈选出图像中需要修复的部位,手动输入修复内容并挑选生成效果最好的扩散生成模型,此过程需要不断调整修复内容及大量的模型参数才能达到较好的生成效果。
2、由此可见,现有的图像修复方案操作繁琐、耗时长且无法实现自动化修复。
技术实现思路
1、有鉴于此,本公开实施例提供了一种图像修复方法、装置、计算机设备及存储介质,以解决现有的图像修复方案操作繁琐、耗时长且无法实现自动化修复的问题。
2、第一方面,本公开实施例提供了一种图像修复方法,其特征在于,所述方法包括:
3、获取携带至少一个响应区域的热力图,其中,所述热力图是利用原始图像的原始标签生成的,所述热力图中的每个响应区域对应一种原始标签,所述原始标签是对所述原始图像进行内容理解得到的;
4、获取所述原始标签对应的目标修复引导信息,并获取所述目标修复引导信息对应的目标生成模型;
5、将所述热力图与所述目标修复引导信息输入至所述目标生成模型,通过所述目标生成模型利用所述目标修复引导信息对应的目标特征对所述热力图中的响应区域进行修复,得到目标图像。
6、第二方面,本公开实施例提供了一种图像修复装置,所述装置包括:
7、获取模块,用于获取携带至少一个响应区域的热力图,其中,所述热力图是利用原始图像的原始标签生成的,所述热力图中的每个响应区域对应一种原始标签,所述原始标签是对所述原始图像进行内容理解得到的;
8、处理模块,用于获取所述原始标签对应的目标修复引导信息,并获取所述目标修复引导信息对应的目标生成模型;
9、修复模块,用于将所述热力图与所述目标修复引导信息输入至所述目标生成模型,通过所述目标生成模型利用所述目标修复引导信息对应的目标特征对所述热力图中的响应区域进行修复,得到目标图像。
10、第三方面,本公开实施例提供了一种计算机设备,包括:存储器和处理器,存储器和处理器之间互相通信连接,存储器中存储有计算机指令,处理器通过执行计算机指令,从而执行上述第一方面或其对应的任一实施方式的方法。
11、第四方面,本公开实施例提供了一种计算机可读存储介质,该计算机可读存储介质上存储有计算机指令,计算机指令用于使计算机执行上述第一方面或其对应的任一实施方式的方法。
12、本公开实施例提供的方法利用对原始图像进行内容理解得到的原始标签获取相应的修复引导信息,并利用修复引导信息对应的目标生成模型对热力图中的响应区域进行修复,实现了根据图像的内容理解自动进行内容修复。整个过程无需使用者输入修复内容以及调整模型参数,提高了图像的修复效率。
1.一种图像修复方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述获取携带至少一个响应区域的热力图,包括:
3.根据权利要求2所述的方法,其特征在于,所述从所述原始图像中获取与所述原始标签相匹配的目标图像特征,包括:
4.根据权利要求1所述的方法,其特征在于,所述获取所述原始标签对应的目标修复引导信息,包括:
5.根据权利要求4所述的方法,其特征在于,所述获取所述目标修复引导信息对应的目标生成模型,包括:
6.根据权利要求5所述的方法,其特征在于,所述条件控制网络的训练方法包括:
7.根据权利要求5所述的方法,其特征在于,所述通过所述目标生成模型利用所述目标修复引导信息对应的目标特征对所述热力图中的响应区域进行修复,得到目标图像,包括:
8.一种图像修复装置,其特征在于,所述装置包括:
9.一种计算机设备,其特征在于,包括:
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机指令,所述计算机指令用于使计算机执行权利要求1至7中任一项所述的方法。