[发明专利]图像处理方法、装置、电子设备及存储介质

说明书

本公开关于一种图像处理方法、装置、电子设备及存储介质。该方法包括：获取待处理图像和级联抠图模型，分别根据第一分辨率和大于第一分辨率的第二分辨率对待处理图像进行采样处理，得到第一级联图像和第二级联图像；将第一级联图像输入第一级联网络进行像素属性特征提取处理，得到第一特征图像；根据第二分辨率对第一特征图像进行上采样处理，得到目标上采样特征图像；将目标上采样特征图像和第二级联图像输入第二级联网络进行图像前景提取处理，得到第一目标掩码图像以用于从待处理图像中提取目标前景图像。根据本公开提供的技术方案，可以从低分辨率至高分辨进行抠图，提高抠图精度。

技术领域

本公开涉及计算机视觉技术领域，尤其涉及一种图像处理方法、装置、电子设备及存储介质。

背景技术

随着多媒体的蓬勃发展，对多媒体的处理，例如对图像或视频的背景进行替换等，一般基于抠图技术，实现对图像或视频的背景进行替换。相关技术中，利用三分图方法或全自动抠图方法(基于机器学习的方法)实现对图像或视频的背景进行替换，但三分图方法的抠图效果受三分图质量的影响较大；而全自动抠图方法在前景与背景的特征较接近时，在前景与背景的交界区域易发生误判。因此，三分图方法或全自动抠图方法的抠图精度均较低，不能满足现有抠图需求。

发明内容

本公开提供一种图像处理方法、装置、电子设备及存储介质，以至少解决相关技术中如何提高抠图效率和抠图效果的问题。本公开的技术方案如下：

根据本公开实施例的第一方面，提供一种图像处理方法，包括：

获取待处理图像和级联抠图模型，所述级联抠图模型包括与第一分辨率对应的第一级联网络以及与第二分辨率对应的第二级联网络；所述第二分辨率大于所述第一分辨率；

分别根据所述第一分辨率和所述第二分辨率，对所述待处理图像进行采样处理，得到第一级联图像和第二级联图像；

将所述第一级联图像输入所述第一级联网络进行像素属性特征提取处理，得到第一特征图像；

根据所述第二分辨率对所述第一特征图像进行上采样处理，得到目标上采样特征图像；

将所述目标上采样特征图像和所述第二级联图像输入所述第二级联网络进行图像前景提取处理，得到第一目标掩码图像；

根据所述第一目标掩码图像，从所述待处理图像中提取目标前景图像。

在一种可能的实现方式中，所述级联抠图模型还包括至少一个中间级联网络，每个中间级联网络对应一个中间分辨率；每个中间分辨率大于所述第一分辨率且小于所述第二分辨率；所述至少一个中间级联网络按照对应的中间分辨率从低至高排序，其中，所述第一级联网络与最低中间分辨率对应的中间级联网络连接，所述最高中间分辨率对应的中间级联网络与所述第二级联网络连接；所述图像处理方法还包括依次迭代以下步骤：

获取当前中间级联网络对应的中间级联图像以及对应的目标中间分辨率，所述中间级联图像是基于所述目标中间分辨率对所述待处理图像进行采样处理得到的；

获取所述当前中间级联网络的前一级联网络输出的特征图像；

根据所述目标中间分辨率对所述前一级联网络输出的特征图像进行上采样处理，得到第一上采样特征图像；

将所述中间级联图像和所述第一上采样特征图像作为所述当前中间级联网络的输入，进行像素属性特征提取处理，得到目标特征图像。

在一种可能的实现方式中，所述图像处理方法还包括：

将所述最高中间分辨率对应的中间级联网络的目标特征图像作为中间特征图像；

根据所述第二分辨率对所述中间特征图像进行上采样处理，得到第二上采样特征图像；