[发明专利]图像分割方法、装置、设备及计算机存储介质

说明书

本申请提供了一种图像分割方法、装置、设备及计算机存储介质；方法包括：响应于进行图像分割的操作指令，获取待处理视频中的当前帧图像和前一帧图像的图像分割结果；基于当前帧图像和前一帧图像的图像分割结果确定当前帧图像的预测前景概率结果；基于当前帧图像的预测前景概率结果、前一帧图像至前k帧图像的前景概率结果，确定当前帧图像的目标前景概率结果；基于当前帧图像的目标前景概率结果和当前帧图像，确定当前帧图像的图像分割结果；至少基于当前帧图像的图像分割结果和当前帧图像，确定目标图像；输出并显示目标图像。如此，能够利用前一帧图像的分割结果指导神经网络模型进行当前帧的图像分割，从而得到时序连续的分割结果。

技术领域

本申请涉及人工智能技术领域，尤其涉及一种图像分割方法、装置、设备及计算机存储介质。

背景技术

随着人工智能技术和智能终端的不断发展，人们的工作、生活、娱乐都有了翻天覆地的变化。例如人们可以利用移动终端随时随地与其他人进行即时通讯或者拍摄图片或视频等。为了增加娱乐趣味性，还可以将拍摄的图片或视频进行图像分割以更换图像或视频的背景。

目前娱乐场景下的基于深度学习的图像人像分割技术较为成熟，然而不同于图像分割技术，视频分割除了保证语义区域分割的正确性之外还需确保分割结果同样保持时序连续以确保在娱乐应用场景下不会出现肉眼可见的闪烁。然而由于移动端可支持的计算量有限难以支撑边缘等难分区域的准确分割，同时分割本身的二义性以及训练数据标注的二义性等原因，现有技术方案很难保证视频分割结果的时序连续性，导致分割后的视频常常能看到明显的区域闪烁，或严重的分割残影，极大的影响了用户使用体验。

发明内容

本申请实施例提供一种图像分割方法、装置及计算机存储介质，能够利用前一帧图像的分割结果指导神经网络模型进行当前帧的图像分割，从而得到时序连续的分割结果。

本申请实施例的技术方案是这样实现的：

本申请实施例提供一种图像分割方法，包括：

响应于进行图像分割的操作指令，获取待处理视频中的当前帧图像和该当前帧图像的前一帧图像的图像分割结果；

基于该当前帧图像和前一帧图像的图像分割结果确定该当前帧图像的预测前景概率结果；

基于该当前帧图像的预测前景概率结果、当前帧图像的前一帧图像至前k帧图像的前景概率结果，确定该当前帧图像的目标前景概率结果；

基于该当前帧图像的目标前景概率结果和该当前帧图像，确定该当前帧图像的图像分割结果；

至少基于该当前帧图像的图像分割结果和该当前帧图像，确定目标图像；

输出并显示该目标图像。

本申请实施例提供一种图像分割装置，包括：

第一获取模块，用于响应于进行图像分割的操作指令，获取待处理视频中的当前帧图像和该当前帧图像的前一帧图像的图像分割结果；

第一确定模块，用于基于该当前帧图像和前一帧图像的图像分割结果确定该当前帧图像的预测前景概率结果；

第二确定模块，用于基于该当前帧图像的预测前景概率结果、当前帧图像的前一帧图像至前k帧图像的前景概率结果，确定该当前帧图像的目标前景概率结果；

第三确定模块，用于基于该当前帧图像的目标前景概率结果和该当前帧图像，确定该当前帧图像的图像分割结果；

第四确定模块，用于至少基于该当前帧图像的图像分割结果和该当前帧图像，确定目标图像；

输出模块，用于输出并显示该目标图像。

在一些实施例中，当该当前帧图像为该待处理视频中的第一帧图像时，该图像分割装置还包括：