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

权利要求书

1.一种图像分割方法，包括：

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

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

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

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

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

2.如权利要求1所述的方法，其中，所述根据所述视频图像帧中包含的各候选区域的像素梯度和所述视频图像帧的像素梯度，确定所述视频图像帧中存在运动模糊的目标图像区域，包括：

确定各所述候选图像区域在各个设定方向上的像素梯度；

根据各所述候选图像区域在各个设定方向上的像素梯度，确定各个所述候选图像区域的像素梯度方差；

根据各所述候选图像区域的像素梯度方差，与所述视频图像帧的像素梯度方差的匹配程度，确定所述视频图像帧中存在运动模糊的目标图像区域。

3.如权利要求2所述的方法，其中，所述根据各所述候选图像区域的像素梯度方差，与所述视频图像帧的像素梯度方差的匹配程度，确定所述视频图像帧中存在运动模糊的目标图像区域之前，包括：

获取设定大小的滑动窗口；

将所述滑动窗口在所述视频图像帧中按照设定的步长移动，以确定所述视频图像帧中的多个候选图像区域。

4.如权利要求1所述的方法，其中，所述将所述待分割的视频图像帧，输入图像分割模型，得到目标分割图像，包括：

将所述视频图像帧，输入图像分割模型的分割网络，以得到初始分割图像和误差估计图；所述初始分割图像中各元素用于指示所述视频图像帧中对应像素点的初始分割置信度；误差估计图中各元素用于指示所述视频图像帧中对应像素点的初始分割置信度的估计误差；

将所述视频图像帧、所述误差估计图和所述初始分割图像，输入所述图像分割模型的修正网络，得到所述目标分割图像；所述修正网络根据误差估计图和视频图像帧，输出各像素点的初始分割置信度的实际误差，根据实际误差对所述初始分割图像的初始分割置信度进行修正，得到目标分割图像；目标分割图像中各元素用于指示所述视频图像帧中对应像素点的目标分割置信度。

5.如权利要求1所述的方法，其中，所述根据设定阈值，对所述目标图像区域中各像素点对应的目标分割置信度进行修正，包括：

将所述目标图像区域中各像素点对应的目标分割置信度与设定阈值比较；

将目标分割置信度小于设定阈值的各所述像素点的目标分割置信度设置为设定值。

6.如权利要求1-5任一项所述的方法，其中，所述方法还包括：

获取训练样本图像；所述训练样本图像的背景为设定颜色的纯背景；

根据所述训练样本图像，对所述图像分割模型进行训练。

7.一种图像分割装置，包括：

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

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

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

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

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