[发明专利]样本确定方法及装置、计算机可读存储介质和电子设备

说明书

本公开提供了一种样本确定方法、样本确定装置、计算机可读存储介质和电子设备，涉及图像处理技术领域。该样本确定方法包括：对待标注图像进行主体检测；在检测出待标注图像包含主体的情况下，针对主体对待标注图像进行抠图操作，以生成主体图像和第一掩膜图像；确定主体图像的质量评价结果，在主体图像的质量评价结果满足质量要求的情况下，将待标注图像和所述第一掩膜图像确定为第一正样本，以应用于目标抠图模型的训练过程。本公开可以降低样本数据的获取成本并提高获取效率。

技术领域

本公开涉及图像处理技术领域，具体而言，涉及一种样本确定方法、样本确定装置、计算机可读存储介质和电子设备。

背景技术

应用于移动终端的抠图算法的能力提升很大程度依赖于样本数据。目前，样本数据的获取仍依赖人工手动标注，这就造成样本数据的获取成本高且效率低的问题。

发明内容

本公开提供一种样本确定方法、样本确定装置、计算机可读存储介质和电子设备，进而至少在一定程度上克服移动终端上的抠图模型的样本数据获取成本高且效率低的问题。

根据本公开的第一方面，提供了一种样本确定方法，包括：对待标注图像进行主体检测；在检测出待标注图像包含主体的情况下，针对主体对待标注图像进行抠图操作，以生成主体图像和第一掩膜图像；确定主体图像的质量评价结果，在主体图像的质量评价结果满足质量要求的情况下，将待标注图像和第一掩膜图像确定为第一正样本，以应用于目标抠图模型的训练过程。

根据本公开的第二方面，提供了一种样本确定装置，包括：主体检测模块，用于对待标注图像进行主体检测；图像生成模块，用于在检测出待标注图像包含主体的情况下，针对主体对待标注图像进行抠图操作，以生成主体图像和第一掩膜图像；样本确定模块，用于确定主体图像的质量评价结果，在主体图像的质量评价结果满足质量要求的情况下，将待标注图像和第一掩膜图像确定为第一正样本，以应用于目标抠图模型的训练过程。

根据本公开的第三方面，提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现上述的样本确定方法。

根据本公开的第四方面，提供了一种电子设备，包括处理器；存储器，用于存储一个或多个程序，当一个或多个程序被处理器执行时，使得所述处理器实现上述的样本确定方法。

在本公开的一些实施例所提供的技术方案中，在检测出待标注图像包含主体的情况下，进行抠图操作以生成主体图像，并在主体图像的质量评价结果满足质量要求的情况下，确定第一正样本以训练目标抠图模型。相比于纯人工确定样本数据的方案，本公开方案可以减少人工的参与，人工成本降低的同时，样本数据的获取效率得到了提高，进而有助于缩短抠图模型的迭代周期。

应当理解的是，以上的一般描述和后文的细节描述仅是示例性和解释性的，并不能限制本公开。

附图说明

此处的附图被并入说明书中并构成本说明书的一部分，示出了符合本公开的实施例，并与说明书一起用于解释本公开的原理。显而易见地，下面描述中的附图仅仅是本公开的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。在附图中：

图1示出了本公开实施方式的与移动终端的抠图模型训练相关的系统示意图；

图2示意性示出了根据本公开示例性实施方式的样本确定方法的流程图；

图3以第一正样本为例示出了本公开实施例的样本扩展的示意图；

图4示意性示出了本公开实施例的第一正样本数据池和第二正样本数据池应用于不同训练阶段的示意图；

图5示出了本公开实施例的目标抠图模型的示意图；

图6示意性示出了本公开实施例的样本确定方案的整个过程的流程图；