一种抠图数据生成方法、装置、计算机设备及存储介质与流程

技术特征：
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
＝∑
j
a
ij
；按照下式构建一封闭形式解的方程，并将所述封闭形式解作为所述目标掩膜图像：式中，λ为约束系数。5.根据权利要求1所述的抠图数据生成方法，其特征在于，所述基于交并比计算所述原始掩膜图像和目标掩膜图像的重叠程度，包括：按照下式计算所述原始掩膜图像和目标掩膜图像的交并比，并将计算结果作为所述重
叠程度：式中，a为所述原始掩膜图像，b为所述目标掩膜图像。6.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据，包括：当所述重叠程度在所述预设重叠范围内时，则判定所述目标掩膜图像与所述原始掩膜图像的重叠程度低；当所述重叠程度未在所述预设重叠范围内时，则判定所述目标掩膜图像与所述原始掩膜图像的重叠程度高；将重叠程度低的目标掩膜图像对应的目标图像设置为所述抠图数据。7.根据权利要求1所述的抠图数据生成方法，其特征在于，所述将所述原始掩膜图像和背景图像融合为一目标图像之前，包括：基于预设数据生成策略对所述前景图像和原始掩膜图像进行预处理，并采用预处理后的原始掩膜图像与所述背景图像进行融合；其中，所述预设数据生成策略由随机前景模糊、随机前景旋转角度和颜色转换任意组合得到。8.一种抠图数据生成装置，其特征在于，包括：图像获取单元，用于分别获取前景图像和背景图像，并提取所述前景图像中的原始掩膜图像；第一融合单元，用于将所述原始掩膜图像和背景图像融合为一目标图像；模型输出单元，用于将所述目标图像输入至抠图模型中，并由所述抠图模型输出所述目标图像的目标掩膜图像；第一计算单元，用于基于交并比计算所述原始掩膜图像和目标掩膜图像的重叠程度；数据生成单元，用于将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据。9.一种计算机设备，其特征在于，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如权利要求1至7任一项所述的抠图数据生成方法。10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至7任一项所述的抠图数据生成方法。

技术总结
本发明公开了一种抠图数据生成方法、装置、计算机设备及存储介质，该方法包括：分别获取前景图像和背景图像，并提取前景图像中的原始掩膜图像；将原始掩膜图像和背景图像融合为一目标图像；将目标图像输入至抠图模型中，并由抠图模型输出目标图像的目标掩膜图像；基于交并比计算原始掩膜图像和目标掩膜图像的重叠程度；将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据。本发明实施例通过将前景图像中的原始掩膜图像与背景图像相融合为所述目标图像，然后利用抠图模型对该目标图像进行预测，并基于预测结果判断该目标图像是否有利于抠图模型的训练，以此生成可有效提升抠图模型训练效果的抠图数据。训练效果的抠图数据。训练效果的抠图数据。


技术研发人员：陈信宇
受保护的技术使用者：深圳万兴软件有限公司
技术研发日：2021.12.27
技术公布日：2022/3/25