[发明专利]一种抠图数据生成方法、装置、计算机设备及存储介质

权利要求书

1.一种抠图数据生成方法，其特征在于，包括：

分别获取前景图像和背景图像，并提取所述前景图像中的原始掩膜图像；

将所述原始掩膜图像和背景图像融合为一目标图像；

将所述目标图像输入至抠图模型中，并由所述抠图模型输出所述目标图像的目标掩膜图像；

基于交并比计算所述原始掩膜图像和目标掩膜图像的重叠程度；

将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据。

2.根据权利要求1所述的抠图数据生成方法，其特征在于，所述分别获取前景图像和背景图像，并提取所述前景图像中的原始掩膜图像，包括：

将所述前景图像转换为灰度图像，并隔离所述灰度图像的边缘像素；

对隔离后的灰度图像进行反转，并创建掩码；

通过位运算提取所述灰度图像的边界信息，并基于所述边界信息对所述灰度图像进行叠加，以此提取得到所述原始掩膜图像。

3.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将所述原始掩膜图像和背景图像融合为一目标图像，包括：

按照下式对所述原始掩膜图像和背景图像进行融合：

M＝I*α+B(1-α)

式中，M为所述目标图像，I为所述前景图，α为所述原始掩膜图像，B为所述背景图。

4.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将所述目标图像输入至抠图模型中，并由所述抠图模型输出所述目标图像的目标掩膜图像，包括：

按照下式，计算所述目标掩膜图像中的像素i的特征向量：

X(i)＝(cos(h),sin(h),s,v,x,y)

式中，h、s、v分别是HSV颜色空间的坐标值，(x，y)是像素i的空间坐标；

按照下式设置内核函数：

式中，C为权值调节系数，以保证内核函数k(i，j)∈[0,1]，||·||为1范数；

基于所述内核函数计算拉普拉斯矩阵L：

L＝D-A

式中，相似矩阵A_ij＝k(i，j)，对角矩阵D_ii＝∑_jA_ij；

按照下式构建一封闭形式解的方程，并将所述封闭形式解作为所述目标掩膜图像：

式中，λ为约束系数。

5.根据权利要求1所述的抠图数据生成方法，其特征在于，所述基于交并比计算所述原始掩膜图像和目标掩膜图像的重叠程度，包括：

按照下式计算所述原始掩膜图像和目标掩膜图像的交并比，并将计算结果作为所述重叠程度：

式中，A为所述原始掩膜图像，B为所述目标掩膜图像。

6.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据，包括：

当所述重叠程度在所述预设重叠范围内时，则判定所述目标掩膜图像与所述原始掩膜图像的重叠程度低；

当所述重叠程度未在所述预设重叠范围内时，则判定所述目标掩膜图像与所述原始掩膜图像的重叠程度高；

将重叠程度低的目标掩膜图像对应的目标图像设置为所述抠图数据。

7.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将所述原始掩膜图像和背景图像融合为一目标图像之前，包括：

基于预设数据生成策略对所述前景图像和原始掩膜图像进行预处理，并采用预处理后的原始掩膜图像与所述背景图像进行融合；其中，所述预设数据生成策略由随机前景模糊、随机前景旋转角度和颜色转换任意组合得到。