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

说明书

本发明公开了一种图像处理方法、装置、设备和存储介质。其中，图像处理方法包括：获取当前视频流，当前视频流包括至少一帧包含目标对象图像的待处理视频帧；将待处理视频帧输入至蒙版处理模型中以得到前景蒙版，前景蒙版对应目标对象图像，蒙版处理模型用于通过背景解码子单元和前景解码子单元分别输出待处理视频帧的前景特征图和背景特征图，根据前景特征图和背景特征图确定前景蒙版；根据前景蒙版对待处理视频帧进行背景替换以更新当前视频流中的待处理视频帧；将更新后的当前视频流发送至服务端，以由服务端下发至目标用户端。本公开实施例提供的图像处理方法解决了现有技术的前景人像确定方法所确定的前景人像质量较低的问题。

技术领域

本公开实施例涉及计算机数据处理技术，尤其涉及一种图像处理方法、装置、设备和存储介质。

背景技术

随着直播的流行，人像背景替换的需求越来越大。人像背景替换是指从含有人像的自然图像或视频中精确地确定出前景人像，并将其与预设的背景图像融合起来。以生鲜货品的直播带货为例，可以通过人像背景替换，将主播与展示生鲜货品生产环境的视频融合在一起，这样既能增加直播的多样性和真实性，又能使观众看到主播的人像，从而提升观众购买相关货品的欲望。因此，要进行人像背景替换就要先从含有人像的自然图像或视频中精确地确定出前景人像。

现有前景人像确定方法主要包括语义分割和人像抠图，前者的缺点是边缘缺失严重，粒度较粗；后者虽然人像边缘细节保留的较好，但整体性较差。由此可见，现有技术的前景人像确定方法所确定的前景人像的质量较低，通常无法满足人们的审美要求。

发明内容

本公开实施例提供一种图像处理方法、装置、设备和存储介质，以解决现有技术的前景人像确定方法所确定的前景人像质量较低的问题。

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

获取当前视频流，所述当前视频流包括至少一帧包含目标对象图像的待处理视频帧；

将所述待处理视频帧输入至蒙版处理模型中以得到前景蒙版，所述前景蒙版对应所述目标对象图像，所述蒙版处理模型用于通过背景解码子单元和前景解码子单元分别输出所述待处理视频帧的前景特征图和背景特征图，以及根据所述前景特征图和背景特征图确定所述前景蒙版；

根据所述前景蒙版对所述待处理视频帧进行背景替换以更新所述当前视频流中的待处理视频帧；

将更新后的当前视频流发送至服务端，以由服务端下发至目标用户端。

第二方面，本公开实施例还提供了一种图像处理装置，包括：

视频获取模块，用于获取当前视频流，所述当前视频流包括至少一帧包含目标对象图像的待处理视频帧；

蒙版模块，用于将所述待处理视频帧输入至蒙版处理模型中以得到前景蒙版，所述前景蒙版对应所述目标对象图像，所述蒙版处理模型用于通过背景解码子单元和前景解码子单元分别输出所述待处理视频帧的前景特征图和背景特征图，以及根据所述前景特征图和背景特征图确定所述前景蒙版；

背景替换模块，用于根据所述前景蒙版对所述待处理视频帧进行背景替换以更新所述当前视频流中的待处理视频帧；

输出模块，用于将更新后的当前视频流发送至服务端，以由服务端下发至目标用户端。

第三方面，本公开实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如任一本公开实施例所述的图像处理方法。