[发明专利]图像分割方法、装置、电子设备和存储介质

说明书

本公开提出一种图像分割方法、装置、电子设备和存储介质，涉及人工智能技术领域，具体为计算机视觉和深度学习技术领域，可应用于图像处理和图像识别场景。具体方案为：将待分割的视频图像帧输入图像分割模型，得到目标分割图像，目标分割图像中各元素指示视频图像帧中对应像素点属于前景物体的目标分割置信度，根据视频图像帧中的各候选区域的像素梯度和视频图像帧的像素梯度，确定视频图像帧中存在运动模糊的目标图像区域，根据设定阈值，对目标图像区域中各像素点对应的目标分割置信度进行修正，根据修正后的目标分割图像，对视频图像帧进行图像分割，根据梯度变化监测出运动模糊区域，并对其分割结果进行修正，提升了图像分割的准确度。

技术领域

本公开涉及人工智能技术领域，具体为计算机视觉和深度学习技术领域，可应用于图像处理和图像识别场景，尤其涉及图像分割方法、装置、电子设备和存储介质。

背景技术

图像分割，例如，人像分割是图像/视频编辑领域的基础技术之一，其作用是将 人像前景与背景分离，可以应用在视频背景切换等场景中。

但是对于视频图像，尤其是存在人物快速移动的视频图像，由于成像设备的限制，图像中会存在运动模糊现象，导致图像分割边缘存在杂色，图像分割的效果较差。

发明内容

本公开提供了一种图像分割方法、装置、电子设备和存储介质。

根据本公开的一方面，提供了一种图像分割方法，包括：

获取待分割的视频图像帧；

将所述待分割的视频图像帧，输入图像分割模型，得到目标分割图像；其中，所 述目标分割图像中各元素用于指示所述视频图像帧中对应像素点属于前景物体的目 标分割置信度；

根据所述视频图像帧中包含的各候选区域的像素梯度和所述视频图像帧的像素梯度，确定所述视频图像帧中存在运动模糊的目标图像区域；

根据设定阈值，对所述目标图像区域中各像素点对应的目标分割置信度进行修正；

根据修正后的目标分割图像，对所述视频图像帧进行图像分割。

根据本公开的另一方面，提供了一种图像分割装置，包括：

获取模块，用于获取待分割的视频图像帧；

识别模块，用于将所述待分割的视频图像帧，输入图像分割模型，得到目标分割图像； 其中，所述目标分割图像中各元素用于指示所述视频图像帧中对应像素点属于前景物体的 目标分割置信度；

确定模块，用于根据所述视频图像帧中包含的各候选区域的像素梯度和所述视频图像 帧的像素梯度，确定所述视频图像帧中存在运动模糊的目标图像区域；

修正模块，用于根据设定阈值，对所述目标图像区域中各像素点对应的目标分割置信 度进行修正；

分割模块，用于根据修正后的目标分割图像，对所述视频图像帧进行图像分割。

根据本公开的另一方面，提供了一种电子设备，包括：

至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执 行，以使所述至少一个处理器能够执行前述一方面所述方法的步骤。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行前述一方面所述方法的步骤。

根据本公开的另一方面，提供了一种计算机程序产品，包括计算机指令，所述计算机指令在被处理器执行时实现前述一方面所述方法的步骤。