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

说明书

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

技术领域

本发明涉及图像处理技术领域，特别涉及一种抠图数据生成方法、装置、计算机设备及存储介质。

背景技术

当前，在对网络模型(例如抠图模型)进行训练时，当训练数据增加到一定数量后,如果数据场景的分布不够均匀,便会造成训练出来的模型有盲点。并且，如果训练数据均是从网络上搜集而来,那么训练数据的分布便会偏向某些特定的场景,例如训练数据中的办公室的场景较少，从而造成抠图模型在办公室内的环境效果下便不好。而在原本数据量很大的状况下,如果要用人力来专门搜集特定场景的数据的成本就会大大的提高。因此，如何解决训练数据不平均，而导致模型抠图效果不好的问题是本领域技术人员需要克服的问题。

发明内容

本发明实施例提供了一种抠图数据生成方法、装置、计算机设备及存储介质，旨在生成可有效提升抠图模型训练效果的抠图数据。

第一方面，本发明实施例提供了一种抠图数据生成方法，包括：

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

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

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

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

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

第二方面，本发明实施例提供了一种抠图数据生成装置，包括：

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

第一融合单元，用于将所述原始掩膜图像和背景图像融合为一目标图像；

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

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

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

第三方面，本发明实施例提供了一种计算机设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如第一方面所述的抠图数据生成方法。

第四方面，本发明实施例提供了一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如第一方面所述的抠图数据生成方法。