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

说明书

本申请提供了一种图像分割方法、装置、电子设备及可读存储介质。所述方法包括：以视频流中的每一帧图像为当前帧图像，确定当前帧图像是检测帧图像还是跟踪帧图像；在当前帧图像是跟踪帧图像的情况下，根据为当前帧图像预先存储的参考裁剪区域和用于框选当前帧图像的前一帧图像中目标物的外接矩形，确定当前帧图像的裁剪区域；根据当前帧图像的裁剪区域，对当前帧图像进行裁剪，得到当前帧图像的局部图像；将当前帧图像的局部图像输入目标物分割模型，得到当前帧图像的局部图像对应的掩膜并从当前帧图像中分割出目标物。通过本申请的方法，可通过逐帧跟踪增大有效目标物在输入模型的图像中的占比来提升精度，以实现图像分割速度与精度的兼得。

技术领域

本申请实施例涉及图像处理领域，尤其涉及一种图像分割方法、装置、电子设备及可读存储介质。

背景技术

在当前快节奏生活的影响下，随着技术的发展，短视频与直播内容的量越来越多，涉及到的领域也越来越广泛，基于视频的目标物分割需求也愈来愈多。目标物分割任务通常用于目标物抠图、直播背景替换、智能摄像等领域。对于视频分割任务，对图像处理的稳定性和处理速度都有很高的要求，而当前传统的图像分割方法无法同时兼顾速度与精度。

发明内容

本申请提供一种图像分割方法、装置、电子设备及可读存储介质，通过逐帧跟踪增大有效目标物在输入模型的图像中的占比来提升精度，从而实现图像分割速度与精度的兼得。

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

以视频流中的每一帧图像为当前帧图像，确定所述当前帧图像是检测帧图像还是跟踪帧图像；

在所述当前帧图像是跟踪帧图像的情况下，根据为所述当前帧图像预先存储的参考裁剪区域和用于框选所述当前帧图像的前一帧图像中目标物的外接矩形，确定所述当前帧图像的裁剪区域；其中，所述为所述当前帧图像预先存储的参考裁剪区域是所述当前帧图像的前一帧图像的裁剪区域；

根据所述当前帧图像的裁剪区域，对所述当前帧图像进行裁剪，得到所述当前帧图像的局部图像；

将所述当前帧图像的局部图像输入目标物分割模型，得到所述当前帧图像的局部图像对应的掩膜；

根据所述当前帧图像的局部图像对应的掩膜，从所述当前帧图像中分割出所述目标物。

可选的，所述方法还包括：

在所述当前帧图像是视频流中的第一帧图像的情况下，或，在所述当前帧图像及所述当前帧图像的前一帧图像均为检测帧图像的情况下，

将所述当前帧图像输入所述目标物分割模型，得到所述当前帧图像对应的掩膜；

根据所述当前帧图像中目标物所在的区域，生成用于框选所述当前帧图像中目标物的外接矩形；

根据所述用于框选所述当前帧图像中目标物的外接矩形，得到所述当前帧图像的裁剪区域；

存储所述当前帧图像的裁剪区域，以作为所述当前帧图像的下一帧图像的参考裁剪区域。

可选的，所述方法还包括：

在所述当前帧图像为检测帧图像，且所述当前帧图像的前一帧图像为跟踪帧图像情况下，

将所述当前帧图像输入所述目标物分割模型，得到所述当前帧图像对应的掩膜；

根据所述当前帧图像对应的掩膜，确定所述当前帧图像中目标物所在的区域；

将所述当前帧图像的前一帧图像中目标物所在的区域的有效像素的数量与所述当前帧图像中目标物所在的区域的有效像素的数量进行对比，确定是否满足滞留条件；