[发明专利]背景替换方法、装置、设备及存储介质

权利要求书

1.一种背景替换方法，其特征在于，所述背景替换方法包括：

在视频帧数据中确定人体感兴趣区域；

基于所述人体感兴趣区域通过预设人像分割网络确定所述人体感兴趣区域中的初始人像掩膜，其中，所述初始人像掩膜包括各像素点的像素概率值；

根据所述像素概率值对所述初始人像掩膜进行图像区域划分，获得目标掩膜；

对所述目标掩膜进行光流跟踪，获取最终掩膜；

根据所述最终掩膜将预设替换背景与所述视频帧数据进行融合，得到背景替换后的目标视频帧。

2.如权利要求1所述的背景替换方法，其特征在于，所述根据所述像素概率值对所述初始人像掩膜进行图像区域划分，获得目标掩膜，包括：

根据所述像素概率值将所述初始人像掩膜划分为前景区域、背景区域以及羽化过渡区域，并将所述前景区域、背景区域以及羽化过渡区域作为当前掩膜；

对所述前景区域以及所述羽化过渡区域进行遍历，确定若干连通区域；

确定各连通区域的面积；

若目标连通区域的面积小于预设面积值，则将所述目标连通区域内各像素点的像素概率值调整为背景概率值，得到目标掩膜。

3.如权利要求2所述的背景替换方法，其特征在于，所述根据所述像素概率值将所述初始人像掩膜划分为前景区域、背景区域以及羽化过渡区域，并将所述前景区域、背景区域以及羽化过渡区域作为当前掩膜，包括：

若第一像素点的像素概率值大于第一预设概率值，则将所述第一像素点划分至前景区域，并将所述第一像素点的像素概率值调整为前景概率值；

若所述第一像素点的像素概率值小于第二预设概率值，则将所述第一像素点划分至背景区域，并将所述第一像素点的像素概率值调整为背景概率值；

若所述第一像素点的像素概率值大于或等于所述第二预设概率值且小于或等于所述第一预设概率值，则将所述第一像素点划分至羽化过渡区域；

将所述前景区域、背景区域以及羽化过渡区域作为当前掩膜。

4.如权利要求1所述的背景替换方法，其特征在于，所述在视频帧数据中确定人体感兴趣区域，包括：

根据预设人体检测算法对视频帧数据进行人体检测，得到若干检测框；

根据所述若干检测框坐标生成目标框；

根据所述目标框确定人体感兴趣区域数据。

5.如权利要求4所述的背景替换方法，其特征在于，所述根据所述若干检测框坐标生成目标框，包括：

获取所述若干检测框的坐标值；

从所述若干检测框的坐标值选择最大横坐标值、最大纵坐标值、最小横坐标值以及最小纵坐标值；

根据所述最大横坐标值、所述最大纵坐标值、所述最小横坐标值以及所述最小纵坐标值生成目标框。

6.如权利要求5所述的背景替换方法，其特征在于，所述根据所述最大横坐标值、所述最大纵坐标值、所述最小横坐标值以及所述最小纵坐标值生成目标框之后，所述方法还包括：

获取预设长宽扩展比例；

根据所述预设长宽扩展比例对所述目标框进行扩展，得到目标检测框；

根据所述目标检测框确定人体感兴趣区域数据。

7.如权利要求1所述的背景替换方法，其特征在于，所述对所述目标掩膜进行光流跟踪，获取最终掩膜，包括：

对所述目标掩膜进行光流跟踪，得到运动数据；

根据所述运动数据对所述目标掩膜进行调整，得到最终掩膜。

8.一种背景替换装置，其特征在于，所述背景替换装置包括：

确定模块，用于在视频帧数据中确定人体感兴趣区域；

分割模块，用于基于所述人体感兴趣区域通过预设人像分割网络确定所述人体感兴趣区域中的初始人像掩膜，其中，所述初始人像掩膜包括各像素点的像素概率值；

划分模块，用于根据所述像素概率值对所述初始人像掩膜进行图像区域划分，获得目标掩膜；

调整模块，用于对所述目标掩膜进行光流跟踪，获取最终掩膜；

替换模块，用于根据所述最终掩膜将预设替换背景与所述视频帧数据进行融合，得到背景替换后的目标视频帧。