本公开实施例涉及图像前景与背景分割技术,尤其涉及一种图像处理方法、装置、电子设备及存储介质。
背景技术:
1、对于抠图场景,对原始图像进行抠图处理可以是将前景物体从原始图像中分离出来,从而可以对前景物体进行特定编辑处理。
2、目前,采用图像分割模型检测原始图像中前景物体的边缘,并基于边缘检测结果预测前景遮罩。由于前景遮罩的边缘可能存在非平滑的区域,需要对前景遮罩的边缘进行平滑处理等后处理,以提升前景遮罩边缘的平滑程度。然而,平滑处理会破坏前景遮罩的边缘细节信息,进而影响抠图效果。
技术实现思路
1、本公开提供一种图像处理方法、装置、电子设备及存储介质,可以在对前景遮罩的后处理过程中保护边缘细节信息。
2、第一方面,本公开实施例提供了一种图像处理方法,包括:
3、获取原始遮罩图、平滑参数和迭代次数,基于所述原始遮罩图、平滑参数和迭代次数执行平滑处理,得到平滑遮罩图;其中,所述迭代次数表征结束平滑处理的目标次数;
4、根据所述平滑遮罩图相对于原始遮罩图的边缘位移信息确定细节区域;
5、根据所述细节区域生成权重图,根据所述权重图融合所述原始遮罩图和平滑遮罩图得到目标遮罩图,所述目标遮罩图用于执行抠图处理。
6、第二方面,本公开实施例还提供了一种图像处理装置,包括:
7、平滑处理模块,用于获取原始遮罩图、平滑参数和迭代次数,基于所述原始遮罩图、平滑参数和迭代次数执行平滑处理,得到平滑遮罩图;其中,所述迭代次数表征结束平滑处理的目标次数;
8、区域确定模块,用于根据所述平滑遮罩图相对于原始遮罩图的边缘位移信息确定细节区域;
9、遮罩确定模块,用于根据所述细节区域生成权重图,根据所述权重图融合所述原始遮罩图和平滑遮罩图得到目标遮罩图,所述目标遮罩图用于执行抠图处理。
10、第三方面,本公开实施例还提供了一种电子设备,所述电子设备包括:
11、一个或多个处理器;
12、存储装置,用于存储一个或多个程序,
13、当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现如本公开任意实施例所述的图像处理方法。
14、第四方面,本公开实施例还提供了一种包含计算机可执行指令的存储介质,所述计算机可执行指令在由计算机处理器执行时用于执行本公开任意实施例所述的图像处理方法。
15、本公开实施例提供了一种图像处理方法、装置、电子设备及存储介质,通过基于平滑参数和原始遮罩图迭代执行平滑处理,并根据迭代次数控制平滑强度,得到平滑遮罩图;根据平滑遮罩图和原始遮罩图确定因平滑处理破坏掉的细节区域,采用基于细节区域生成的权重图融合原始遮罩图和平滑遮罩图得到目标遮罩图。本公开实施例解决平滑处理破坏前景遮罩的边缘细节信息的问题,由于权重图可以表征需要保护的原始遮罩中的边缘细节信息,基于权重图融合原始遮罩图和平滑遮罩图,可以实现兼顾前景边缘的平滑效果以及对边缘细节信息的保护,提升了抠图效果。
1.一种图像处理方法,其特征在于,包括:
2.根据权利要求1所述的方法,其特征在于,所述基于所述原始遮罩图、平滑参数和迭代次数执行平滑处理,得到平滑遮罩图,包括:
3.根据权利要求2所述的方法,其特征在于,所述按照所述迭代次数基于所述平滑参数和原始遮罩缩小图迭代执行平滑处理,得到候选平滑遮罩图,包括:
4.根据权利要求1所述的方法,其特征在于,所述根据所述细节区域生成权重图,包括:
5.根据权利要求4所述的方法,其特征在于,所述根据所述平滑遮罩图相对于原始遮罩图的边缘位移信息确定细节区域,包括:
6.根据权利要求5所述的方法,其特征在于,所述根据所述边缘凸出区域和边缘凹陷区域确定细节区域,包括:
7.根据权利要求5所述的方法,其特征在于,所述对所述平滑遮罩图进行腐蚀处理得到腐蚀遮罩图,对所述平滑遮罩图进行膨胀处理得到膨胀遮罩图,包括:
8.根据权利要求7所述的方法,其特征在于,所述根据所述权重图融合所述原始遮罩图和平滑遮罩图得到目标遮罩图,包括:
9.根据权利要求8所述的方法,其特征在于,所述根据所述权重系数融合所述原始遮罩图和平滑遮罩图得到目标遮罩图,包括:
10.一种图像处理装置,其特征在于,包括:
11.一种电子设备,其特征在于,所述电子设备包括:
12.一种包含计算机可执行指令的存储介质,其特征在于,所述计算机可执行指令在由计算机处理器执行时用于执行如权利要求1-9中任一所述的图像处理方法。