(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111613442.X (22)申请日 2021.12.27 (71)申请人 深圳万兴软件 有限公司 地址 518000 广东省深圳市南 山区粤海街 道软件产业基地5 栋D座1001 (72)发明人 陈信宇　 (74)专利代理 机构 深圳市精英专利事务所 44242 代理人 武志峰 (51)Int.Cl. G06T 5/00(2006.01) G06T 5/50(2006.01) G06T 7/194(2017.01) G06V 10/774(2022.01) G06K 9/62(2022.01) (54)发明名称 一种抠图数据生成方法、 装置、 计算机设备 及存储介质 (57)摘要 本发明公开了一种抠图数据生成方法、 装 置、 计算机设备及存储介质， 该方法包括： 分别获 取前景图像和背景图像， 并提取前景图像中的原 始掩膜图像； 将原始掩膜图像和背景图像融合为 一目标图像； 将目标图像输入至抠图模型中， 并 由抠图模型输出目标图像的目标掩膜图像； 基于 交并比计算原始掩膜 图像和目标掩膜 图像的重 叠程度； 将 重叠程度在预设重叠范围内的目标掩 膜图像所对应的目标图像作为用于训练抠图模 型的抠图数据。 本发明实施例通过将前景图像中 的原始掩膜 图像与背景图像相融合为所述目标 图像， 然后利用抠图模型对该目标图像进行预 测， 并基于预测结果判断该目标图像是否有利于 抠图模型的训练， 以此生 成可有效提升抠图模型 训练效果的抠图数据。 权利要求书2页 说明书7页 附图3页 CN 114240804 A 2022.03.25 CN 114240804 A 1.一种抠图数据生成方法， 其特 征在于， 包括： 分别获取 前景图像和背景图像， 并提取 所述前景图像中的原 始掩膜图像； 将所述原 始掩膜图像和背景图像融合 为一目标图像； 将所述目标图像输入至抠图模型中， 并由所述抠图模型输出所述目标图像的目标掩膜 图像； 基于交并比计算所述原 始掩膜图像和目标掩膜图像的重 叠程度； 将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像作为用于训练抠图 模型的抠图数据。 2.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述分别获取前景图像和背 景图像， 并提取 所述前景图像中的原 始掩膜图像， 包括： 将所述前 景图像转换为灰度图像， 并隔离所述灰度图像的边 缘像素； 对隔离后的灰度图像进行反转， 并创建掩码； 通过位运算提取所述灰度图像的边界信 息， 并基于所述边界信 息对所述灰度图像进行 叠加， 以此提取 得到所述原 始掩膜图像。 3.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述将所述原始掩膜图像和 背景图像融合 为一目标图像， 包括： 按照下式对所述原 始掩膜图像和背景图像进行融合： M＝I*α +B(1‑α ) 式中， M为所述目标图像， I 为所述前 景图， α 为所述原 始掩膜图像， B为所述背景图。 4.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述将所述目标图像输入至 抠图模型中， 并由所述 抠图模型输出 所述目标图像的目标掩膜图像， 包括： 按照下式， 计算所述目标掩膜图像中的像素i的特 征向量： X(i)＝(cos(h),si n(h),s,v,x,y) 式中， h、 s、 v分别是HSV颜色空间的坐标值， (x， y)是像素i的空间坐标； 按照下式设置内核函数： 式中， C为权值调节系数， 以保证内核函数k(i， j)∈[0,1]， | |·||为1范数； 基于所述内核函数计算拉普拉斯矩阵L： L＝D‑A 式中， 相似矩阵Aij＝k(i， j)， 对角矩阵Dii＝∑jAij； 按照下式构建一封闭形式解的方程， 并将所述封闭形式解作为所述目标掩膜图像： 式中， λ为约束系数。 5.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述基于交并比计算所述原 始掩膜图像和目标掩膜图像的重 叠程度， 包括： 按照下式计算所述原始掩膜图像和目标掩膜图像的交并比， 并将计算结果作为所述重权　利　要　求　书 1/2 页 2 CN 114240804 A 2叠程度： 式中， A为所述原 始掩膜图像， B为所述目标掩膜图像。 6.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述将重叠程度在预设重叠 范围内的目标掩膜图像所对应的目标图像作为用于训练抠图模型的抠图数据， 包括： 当所述重叠程度在所述预设重叠范围内时， 则判定所述目标掩膜图像与所述原始掩膜 图像的重 叠程度低； 当所述重叠程度 未在所述预设重叠范围内时， 则判定所述目标掩膜图像与 所述原始掩 膜图像的重 叠程度高； 将重叠程度低的目标掩膜图像对应的目标图像设置为所述 抠图数据。 7.根据权利要求1所述的抠图数据生成方法， 其特征在于， 所述将所述原始掩膜图像和 背景图像融合 为一目标图像之前， 包括： 基于预设数据生成策略对所述前景图像和 原始掩膜图像进行预处理， 并采用预处理后 的原始掩膜图像与所述背 景图像进 行融合； 其中， 所述预设数据生成策略由随机前景模糊 、 随机前景旋转角度和颜色转换任意组合得到 。 8.一种抠图数据生成装置， 其特 征在于， 包括： 图像获取单元， 用于分别获取前景图像和背景图像， 并提取所述前景图像中的原始掩 膜图像； 第一融合单 元， 用于将所述原 始掩膜图像和背景图像融合 为一目标图像； 模型输出单元， 用于将所述目标图像输入至抠图模型中， 并由所述抠图模型输出所述 目标图像的目标掩膜图像； 第一计算单 元， 用于基于交并比计算所述原 始掩膜图像和目标掩膜图像的重 叠程度； 数据生成单元， 用于将重叠程度在预设重叠范围内的目标掩膜图像所对应的目标图像 作为用于训练抠图模型的抠图数据。 9.一种计算机设备， 其特征在于， 包括存储器、 处理器及存储在所述存储器上并可在所 述处理器上运行的计算机程序， 所述处理器执行所述计算机程序时实现如权利要求 1至7任 一项所述的抠图数据生成方法。 10.一种计算机可读存储介质， 其特征在于， 所述计算机可读存储介质上存储有计算机 程序， 所述计算机程序被处理器执行时实现如权利要求 1至7任一项 所述的抠图数据生 成方 法。权　利　要　求　书 2/2 页 3 CN 114240804 A 3