图像标注方法、装置、设备和介质与流程

专利2022-06-29  86


本发明实施例涉及图像处理技术,尤其涉及一种图像标注方法、装置、设备和介质。



背景技术:

近年来,随着高性能图像处理器和人工智能深度学习技术的进一步发展,越来越多的用户使用人工智能深度学习技术去进行图像识别,而实现基于人工智能深度学习的图像识别的前提是大量的图像标注数据。

传统的图像标注方法中,由标注员根据个人经验和一些已有规则自行判断后完成标注,需要耗费大量的人力和时间,标注效率低,而图像语义分割作为复杂度较高、耗时较长的一种标注任务,会使标注成本更高、标注效率更低。



技术实现要素:

本发明实施例提供一种图像标注方法、装置、设备和介质,以减低标注成本,提高标注效率。

第一方面,本发明实施例提供了一种图像标注方法,包括:

根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

将所述目标轮廓作为所述待标注图像的标注数据。

第二方面,本发明实施例还提供了一种图像标注装置,该装置包括:

模型预标注模块,用于根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

标注数据调整模块,用于根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

标注数据确认模块,用于将所述目标轮廓作为所述待标注图像的标注数据。

第三方面,本发明实施例还提供了一种计算机设备,所述计算机设备包括:

一个或多个处理器;

存储器,用于存储一个或多个程序,

当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现任意实施例所述的图像标注方法。

第四方面,本发明实施例还提供了一种计算机可读存储介质,其上存储有计算机程序,该程序被处理器执行时实现任意实施例所述的图像标注方法。

本发明实施例中,首先根据预先生成的目标标注模型对待标注图像进行预标注生成掩膜图像,得到初步标注结果,然后根据初步标注结果对待标注图像中目标对象的轮廓进行调整,确定出目标对象的目标轮廓,作为目标标注模型对待标注图像的最终标注结果,进而在提高标注效率的基础上,提升了模型标注结果的准确率,以此也可降低人工对模型标注结果的修改量。

附图说明

图1是本发明实施例一中的一种图像标注方法的流程图;

图2是本发明实施例二中的一种图像标注方法的流程图;

图3是本发明实施例二中的一种图像标注方法中生成的不同待定轮廓的对比示意图;

图4是本发明实施例二中的一种图像标注方法所适用的一种工作流程图;

图5是本发明实施例三中的一种图像标注装置的模块结构示意图;

图6是本发明实施例四中的一种计算机设备的结构示意图。

具体实施方式

下面结合附图和实施例对本发明作进一步的详细说明。可以理解的是,此处所描述的具体实施例仅仅用于解释本发明,而非对本发明的限定。另外还需要说明的是,为了便于描述,附图中仅示出了与本发明相关的部分而非全部结构。

实施例一

图1是本发明实施例一提供的一种图像标注方法的流程图,本实施例可适用于如何生成大量标注复杂度较高、耗时较长的图像语义分割标注样本的情况,该方法可以由本发明任意实施例提供的图像标注装置来执行,该装置可由硬件和/或软件组成,并一般可集成在计算机设备中。

如图1所述,本实施例提供的图像标注方法包括以下步骤:

s110、根据目标标注模型对待标注图像进行预标注,生成掩膜图像。

待标注图像,指的是未进行标注的样本图像。目标标注模型,指的是预先选定的用于对待标注图像进行图像语义分割标注的标注模型,或者是基于大量图像标注样本训练原始机器学习模型生成的用于对待标注图像进行图像语义分割标注的标注模型。

其中,待标注图像中包括目标对象,以及所述目标对象的背景。例如,当待标注图像为人脸图像时,目标对象可以是人脸。图像语义分割标注具体可以指的是,将待标注图像中目标对象的轮廓标注出来,例如是将人脸图像中人脸的轮廓标注出来。

使用目标标注模型对待标注图像进行预标注,即为对待标注图像中的目标对象的轮廓进行标注,同时,基于预标注结果生成掩膜图像。在生成的掩膜图像中,与目标对象轮廓及轮廓内区域对应的像素点的像素值为1,与目标对象轮廓外区域对应的像素点的像素值为0。

在一种可选的实施方式中,在s110之前,还包括:

在标注图像样本中,确定训练样本集和测试样本集;加载至少两种原始模型以及匹配的数据库,并根据所述训练样本集对所述至少两种原始模型进行训练,生成至少两种训练模型;根据所述测试样本集对所述至少两种训练模型分别进行测试,并记录每个所述训练模型的性能指标;将所述性能指标最优的一个训练模型作为所述目标标注模型。

标注图像样本,指的是标注员人工按照实际需求(例如是图像语义分割等)进行标注后的图像。当标注图像样本的数量达到设定阈值时,即可实现对原始模型(例如是机器学习模型)的训练,生成训练模型。

具体的,当标注图像样本的数量达到设定阈值时,可以依据预设规则将标注图像样本划分为训练样本集和测试样本集,例如是在标注图像样本中随机选取总数量的70%作为训练样本集,在标注图像样本中随机选取总数量的30%作为测试样本集。

分别从模型库中加载不同的机器学习模型以及匹配的数据库,使用生成的训练样本集分别对不同的机器学习模型进行训练,生成对应的训练模型。生成训练模型之后,使用测试样本集分别对不同的训练模型进行测试,记录每个训练模型的性能指标,选择出其中性能指标最优的一个训练模型,作为目标标注模型,其中,性能指标具体可以是指标注准确率。

s120、根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓。

轮廓,可以指的是具有至少三个顶点的多边形。掩膜图像中存在所述待标注图像中目标对象的原始轮廓,该原始轮廓为通过目标标注模型对待标注图像进行预标注确定的。

具体的,可以根据所述掩膜图像中的原始轮廓,重新确定出所述目标对象的至少一个待定轮廓。典型的,可以按照预设条件在构成所述原始轮廓的多个顶点中选取一些点作为一个待定轮廓的顶点,以此得到所述目标对象的至少一个待定轮廓。然后,在所述至少一个待定轮廓中依据预设筛选条件选取一个待定轮廓作为目标轮廓,其中,预设筛选条件可以是与轮廓平滑度、顶点数量、轮廓内区域面积等因素相关的筛选条件。

s130、将所述目标轮廓作为所述待标注图像的标注数据。

在本实施例中,目标轮廓即为目标标注模型对待标注图像的自动标注结果。

进一步的,在一种可选的实施方式中,将所述目标轮廓作为所述待标注图像的标注数据之后,还包括:

针对所述待标注图像的标注数据发起人工审核请求;接收对所述待标注图像的标注数据的人工确认信息或人工修改信息。

在得到目标轮廓之后,针对所述待标注图像的标注数据发起人工审核请求,将目标轮廓作为目标标注模型对待标注图像的自动标注结果提供给标注员,以使标注员对待标注图像的自动标注结果(也即目标轮廓)进行人工审核,判断目标标注模型对待标注图像的自动标注结果是否恰当。标注员如果确认标注恰当,则点击确认,如果确认标注不恰当,则点击修改,并对目标标注模型的自动标注结果进行修改。

本发明实施例中,首先根据预先生成的目标标注模型对待标注图像进行预标注生成掩膜图像,得到初步标注结果,然后根据初步标注结果对待标注图像中目标对象的轮廓进行调整,确定出目标对象的目标轮廓,作为目标标注模型对待标注图像的最终标注结果,进而在提高标注效率的基础上,提升了模型标注结果的准确率,以此也可降低人工对模型标注结果的修改量。

实施例二

图2是本发明实施例二提供的一种图像标注方法的流程图。本实施例以上述实施例为基础进行具体化,其中,可以将根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓,具体为:

根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并对所述至少一个待定轮廓分别进行评分,将评分最高的待定轮廓作为所述目标轮廓。

进一步的,还可以将根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,具体为:

根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点;

按照预设斜率阈值调整范围调整所述目标斜率阈值,返回执行根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点的操作,直至得到所述目标对象的所有待定轮廓。

如图2所示,本实施例提供的图像标注方法包括以下步骤:

s210、根据目标标注模型对待标注图像进行预标注,生成掩膜图像。

s220、根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点。

目标斜率阈值,是依据经验预设的一个斜率值或者是一个斜率值区间。

所述目标对象的初始轮廓是由多个点构成的,通过初始轮廓中连续两点的连线斜率与所述目标斜率阈值的关系,来确定所述初始轮廓中每个点是否需要保留,所有保留的点构成一个待定轮廓的所有顶点。目标斜率阈值不同时,得到的待定轮廓也不同。

在一种可选的实施方式中,可以将s220具体为:

在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点;

如果所述第一点与所述第二点的连线斜率不存在,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点的操作,直至所述第一点与所述第二点的连线斜率存在,将所述第一点作为一个待定轮廓顶点;

如果所述第一点与所述第二点的连线斜率大于等于所述目标斜率阈值,则将所述第二点作为一个待定轮廓顶点;

如果所述第一点与所述第二点的连线斜率小于所述目标斜率阈值或者确定出一个待定轮廓顶点,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点的操作,直至处理完所述初始轮廓中的所有点,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点。

具体的,在所述掩膜图像中目标对象的初始轮廓中随机选取一个点(假设为n0点)作为第一点,顺时针或者逆时针选取与第一点相邻的点(假设为n1点)作为第二点,根据下述公式计算第一点(n0点)与第二点(n1点)连线的斜率k:

其中,(xi-1,yi-1)是第一点的坐标,(xi,yi)是第二点的坐标。

如果第一点与第二点连线的斜率k不存在(也即为无穷大),则在所述掩膜图像中目标对象的初始轮廓中按照顺时针或者逆时针的顺序依次选取下一个点(也即上述第二点n1点)重新作为第一点,以及与第一点相邻的点(假设为n2点)重新作为第二点,并根据上述公式重新计算第一点(n1点)与第二点(n2点)连线的斜率k,直至第一点与第二点连线的斜率k存在,将此时的第一点保留,作为待定轮廓的一个顶点,然后继续执行在所述掩膜图像中目标对象的初始轮廓中按照顺时针或者逆时针的顺序依次选取下一个点重新作为第一点,以及与第一点相邻的点重新作为第二点,并根据上述公式重新计算第一点与第二点连线的斜率k的操作,直至处理完所述初始轮廓中的所有点。

如果第一点(n0点)与第二点(n1点)连线的斜率k存在,则判断斜率k与目标斜率阈值的大小关系,如果斜率k大于等于所述目标斜率阈值,则将此时的第二点(n1点)保留,作为待定轮廓的一个顶点,然后继续执行在所述掩膜图像中目标对象的初始轮廓中按照顺时针或者逆时针的顺序依次选取下一个点重新作为第一点,以及与第一点相邻的点重新作为第二点,并根据上述公式重新计算第一点与第二点连线的斜率k的操作,直至处理完所述初始轮廓中的所有点;如果斜率k小于所述目标斜率阈值,则继续执行在所述掩膜图像中目标对象的初始轮廓中按照顺时针或者逆时针的顺序依次选取下一个点(也即上述第二点n1点)重新作为第一点,以及与第一点相邻的点(假设为n2点)重新作为第二点,并根据上述公式重新计算第一点(n1点)与第二点(n2点)连线的斜率k的操作,直至处理完所述初始轮廓中的所有点。

当处理完所述掩膜图像中目标对象的初始轮廓中所有点之后,即可得到构成一个待定轮廓的所有轮廓顶点,将这些轮廓顶点连接起来即为根据所述掩膜图像选取的所述待标注图像中目标对象的一个待定轮廓。

s230、按照预设斜率阈值调整范围调整所述目标斜率阈值,返回执行根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点的操作,直至得到所述目标对象的所有待定轮廓。

预设斜率阈值调整范围,是依据经验预设的一个斜率值调整范围。按照预设斜率阈值调整范围调整所述目标斜率阈值,可以是按照一定的调整步长,在预设斜率阈值调整范围重新确定所述目标斜率阈值的取值。

其中,根据一个目标斜率阈值可以唯一确定所述目标对象的一个待定轮廓,也即所述目标对象的待定轮廓的数量与所述目标斜率阈值的总数量相等。

s240、对所述至少一个待定轮廓分别进行评分,将评分最高的待定轮廓作为所述目标轮廓。

在得到每个待定轮廓之后,依据预设的评分规则对每个待定轮廓进行打分,将得分最高的一个待定轮廓作为目标轮廓。其中,初始轮廓可以作为一个待定轮廓进行打分,并参与目标轮廓的筛选。

在一种可选的实施方式中,可以将对所述待定轮廓进行评分,具体为:

根据所述掩膜图像中所述目标对象的初始轮廓与所述待定轮廓的拟合相似度,以及所述待定轮廓的轮廓顶点数量,对所述待定轮廓进行评分。

其中,所述初始轮廓与所述待定轮廓的拟合相似度,可以根据所述初始轮廓内区域和所述待定轮廓内区域的交集面积与并集面积的比值确定;所述待定轮廓的轮廓顶点数量,可以是归一化后的轮廓顶点数量,典型的,可以根据所述初始轮廓的轮廓顶点数量对所述待定轮廓的轮廓顶点数量进行归一化处理。

具体的,可以根据下述公式对待定轮廓进行评分:

其中,λ1为拟合相似度的权重,λ2为归一化后的待定轮廓的轮廓顶点数量的权重,sn为待定轮廓内区域面积,s0为初始轮廓内区域面积,n为构成待定轮廓的轮廓顶点的数量,n0为构成初始轮廓的轮廓顶点的数量。具体的,拟合相似度的权重λ1可以大于归一化后的待定轮廓的轮廓顶点数量的权重λ2。

针对同一张待标注图像进行标注时,例如是人脸图像,需要基于图像语义分割标注人脸轮廓,图3示出了三种不同的待定轮廓,不同待定轮廓的轮廓顶点的数量n不同。当n过小时,待定轮廓拟合区域重复部分差异较大,当n过大时,待定轮廓拟合区域重复部分差异较小,但标注数据存储量增大,如果标注数据不准确,标注员人工修改量也会增大。

s250、将所述目标轮廓作为所述待标注图像的标注数据。

s260、针对所述待标注图像的标注数据发起人工审核请求,并接收对所述待标注图像的标注数据的人工确认信息或人工修改信息。

标注员获取携带标注数据的待标注图像之后,仅需作简单判断或者微调,不需要人工对每一张待标注图像进行标注,明显提高了标注效率。当目标标注模型以及目标轮廓选取适当时,经实践检验上述技术方案能将标注效率提高几倍甚至是几十倍。

图4示出了上述技术方案所适用的一种工作流程图。如图4所示,标注员从服务器上获取未标注的样本图像,按照实际需求对样本进行标注并将带有标注信息的样本图像提交到服务器;服务器在接收到提交请求后启动一个异步线程,该异步线程检测当前标注任务已标注的样本数量是否达到设置数量阈值,若未达到设定数量阈值则退出异步线程,主线程循环“标注员标注提交,服务器启动异步线程检测样本数量”的过程;随着标注样本图像数量的增加,异步线程检测到当前标注任务已标注的样本图像数量达到设置数量阈值时,异步线程获取到已标注的所有样本图像并生成训练样本集和测试样本集;同时分别从模型库中加载不同的原始模型和数据库,基于训练样本集分别启动训练任务,生成多个训练模型;分别基于测试样本集对多个训练模型进行测试,记录每个训练模型的性能指标;在多个训练模型中选择性能最优的一个训练模型作为目标标注模型,目标标注模型从服务器上获取未标注的样本图像(也即待标注图像)进行预标注,生成mask图像;根据mask图像计算待标注图像中目标对象的初始面积s0以及目标对象初始轮廓包含像素点的数量n0,同时根据mask图像调整所述目标对象的轮廓,得到所述目标对象的至少一个待定轮廓(方法如上所述),并根据待定轮廓内区域的面积sn、待定轮廓包含像素点的数量n、目标对象的初始面积s0以及目标对象初始轮廓包含像素点的数量n0,对每个待定轮廓进行评分(方法如上所述),将评分最高的一个待定轮廓作为目标轮廓,并将所述目标轮廓的轮廓顶点(也即带有标注信息的样本图像)写回服务器;标注员从服务器获取标注有所述目标轮廓的样本图像,判断目标轮廓标注得是否准确,若准确向服务器提交确认信息,若不准确则进行适当修改后提交到服务器。

本实施例未尽详细解释之处请参见前述实施例,在此不再赘述。

上述技术方案,不仅提高了标注效率,也提升了模型标注结果的准确率,进而能够降低人工对模型标注结果的修改量。

实施例三

图5是本发明实施例三提供的一种图像标注装置的模块结构示意图,本实施例可适用于如何生成大量标注复杂度较高、耗时较长的图像语义分割标注样本的情况,该装置可以采用软件和/或硬件的方式实现,并一般可集成在计算机设备中。如图5所示,该装置包括:模型预标注模块510、标注数据调整模块520和标注数据确认模块530,其中,

模型预标注模块510,用于根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

标注数据调整模块520,用于根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

标注数据确认模块530,用于将所述目标轮廓作为所述待标注图像的标注数据。

本发明实施例中,首先根据预先生成的目标标注模型对待标注图像进行预标注生成掩膜图像,得到初步标注结果,然后根据初步标注结果对待标注图像中目标对象的轮廓进行调整,确定出目标对象的目标轮廓,作为目标标注模型对待标注图像的最终标注结果,进而在提高标注效率的基础上,提升了模型标注结果的准确率,以此也可降低人工对模型标注结果的修改量。

进一步的,标注数据调整模块520,具体用于根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并对所述至少一个待定轮廓分别进行评分,将评分最高的待定轮廓作为所述目标轮廓。

进一步的,标注数据调整模块520,具体用于根据所述掩膜图像中所述目标对象的初始轮廓与所述待定轮廓的拟合相似度,以及所述待定轮廓的轮廓顶点数量,对所述待定轮廓进行评分。

进一步的,标注数据调整模块520,具体包括:待定轮廓生成单元和循环单元,其中,

待定轮廓生成单元,用于根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点;

循环单元,用于按照预设斜率阈值调整范围调整所述目标斜率阈值,返回执行根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点的操作,直至得到所述目标对象的所有待定轮廓。

进一步的,待定轮廓生成单元,具体用于在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点;如果所述第一点与所述第二点的连线斜率不存在,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点的操作,直至所述第一点与所述第二点的连线斜率存在,将所述第一点作为一个待定轮廓顶点;如果所述第一点与所述第二点的连线斜率大于等于所述目标斜率阈值,则将所述第二点作为一个待定轮廓顶点;如果所述第一点与所述第二点的连线斜率小于所述目标斜率阈值或者确定出一个待定轮廓顶点,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点的操作,直至处理完所述初始轮廓中的所有点,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点。

进一步的,上述装置还包括:模型生成模块,用于在所述根据目标标注模型对待标注图像进行预标注之前,在标注图像样本中,确定训练样本集和测试样本集;加载至少两种原始模型以及匹配的数据库,并根据所述训练样本集对所述至少两种原始模型进行训练,生成至少两种训练模型;根据所述测试样本集对所述至少两种训练模型分别进行测试,并记录每个所述训练模型的性能指标;将所述性能指标最优的一个训练模型作为所述目标标注模型。

进一步的,上述装置还包括:人工核验模块,用于在所述将所述目标轮廓作为所述待标注图像的标注数据之后,针对所述待标注图像的标注数据发起人工审核请求;接收对所述待标注图像的标注数据的人工确认信息或人工修改信息。

本发明实施例所提供的图像标注装置可执行本发明任意实施例所提供的图像标注方法,具备执行方法相应的功能模块和有益效果。

实施例四

图6是本发明实施例四提供的一种计算机设备的结构示意图,如图6所示,该计算机设备包括处理器60、存储器61、输入装置62和输出装置63;计算机设备中处理器60的数量可以是一个或多个,图6中以一个处理器60为例;计算机设备中的处理器60、存储器61、输入装置62和输出装置63可以通过总线或其他方式连接,图6中以通过总线连接为例。

存储器61作为一种计算机可读存储介质,可用于存储软件程序、计算机可执行程序以及模块,如本发明实施例中的图像标注方法对应的程序指令/模块(例如,附图5所示的图像标注装置中的模型预标注模块510、标注数据调整模块520和标注数据确认模块530)。处理器60通过运行存储在存储器61中的软件程序、指令以及模块,从而执行计算机设备的各种功能应用以及数据处理,即实现上述的图像标注方法。

存储器61可主要包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需的应用程序;存储数据区可存储根据计算机设备的使用所创建的数据等。此外,存储器61可以包括高速随机存取存储器,还可以包括非易失性存储器,例如至少一个磁盘存储器件、闪存器件、或其他非易失性固态存储器件。在一些实例中,存储器61可进一步包括相对于处理器60远程设置的存储器,这些远程存储器可以通过网络连接至计算机设备。上述网络的实例包括但不限于互联网、企业内部网、局域网、移动通信网及其组合。

输入装置62可用于接收输入的数字或字符信息,以及产生与计算机设备的用户设置以及功能控制有关的键信号输入。输出装置63可包括显示屏等显示设备。

实施例五

本发明实施例五还提供一种存储有计算机程序的计算机可读存储介质,计算机程序在由计算机处理器执行时用于执行一种图像标注方法,该方法包括:

根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

将所述目标轮廓作为所述待标注图像的标注数据。

当然,本发明实施例所提供的存储有计算机程序的计算机可读存储介质,其计算机程序不限于如上的方法操作,还可以执行本发明任意实施例所提供的图像标注方法中的相关操作。

通过以上关于实施方式的描述,所属领域的技术人员可以清楚地了解到,本发明可借助软件及必需的通用硬件来实现,当然也可以通过硬件实现,但很多情况下前者是更佳的实施方式。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,如计算机的软盘、只读存储器(read-onlymemory,rom)、随机存取存储器(randomaccessmemory,ram)、闪存(flash)、硬盘或光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本发明各个实施例的方法。

值得注意的是,上述图像标注装置的实施例中,所包括的各个单元和模块只是按照功能逻辑进行划分的,但并不局限于上述的划分,只要能够实现相应的功能即可;另外,各功能单元的具体名称也只是为了便于相互区分,并不用于限制本发明的保护范围。

注意,上述仅为本发明的较佳实施例及所运用技术原理。本领域技术人员会理解,本发明不限于这里的特定实施例,对本领域技术人员来说能够进行各种明显的变化、重新调整和替代而不会脱离本发明的保护范围。因此,虽然通过以上实施例对本发明进行了较为详细的说明,但是本发明不仅仅限于以上实施例,在不脱离本发明构思的情况下,还可以包括更多其他等效实施例,而本发明的范围由所附的权利要求范围决定。


技术特征:

1.一种图像标注方法,其特征在于,包括:

根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

将所述目标轮廓作为所述待标注图像的标注数据。

2.根据权利要求1所述的方法,其特征在于,所述根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓,包括:

根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并对所述至少一个待定轮廓分别进行评分,将评分最高的待定轮廓作为所述目标轮廓。

3.根据权利要求2所述的方法,其特征在于,所述对所述待定轮廓进行评分,包括:

根据所述掩膜图像中所述目标对象的初始轮廓与所述待定轮廓的拟合相似度,以及所述待定轮廓的轮廓顶点数量,对所述待定轮廓进行评分。

4.根据权利要求1所述的方法,其特征在于,所述根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,包括:

根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点;

按照预设斜率阈值调整范围调整所述目标斜率阈值,返回执行根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点的操作,直至得到所述目标对象的所有待定轮廓。

5.根据权利要求4所述的方法,其特征在于,所述根据所述掩膜图像中所述目标对象的初始轮廓中相邻两点的连线斜率与目标斜率阈值的关系,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点,包括:

在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点;

如果所述第一点与所述第二点的连线斜率不存在,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点,并按设定顺序选取与所述第一点相邻的一个第二点的操作,直至所述第一点与所述第二点的连线斜率存在,将所述第一点作为一个待定轮廓顶点;

如果所述第一点与所述第二点的连线斜率大于等于所述目标斜率阈值,则将所述第二点作为一个待定轮廓顶点;

如果所述第一点与所述第二点的连线斜率小于所述目标斜率阈值或者确定出一个待定轮廓顶点,则返回执行在所述掩膜图像中目标对象的初始轮廓中依次选取一个点作为第一点的操作,直至处理完所述初始轮廓中的所有点,得到构成所述目标对象的一个待定轮廓的所有轮廓顶点。

6.根据权利要求1-5任一项所述的方法,其特征在于,在所述根据目标标注模型对待标注图像进行预标注之前,还包括:

在标注图像样本中,确定训练样本集和测试样本集;

加载至少两种原始模型以及匹配的数据库,并根据所述训练样本集对所述至少两种原始模型进行训练,生成至少两种训练模型;

根据所述测试样本集对所述至少两种训练模型分别进行测试,并记录每个所述训练模型的性能指标;

将所述性能指标最优的一个训练模型作为所述目标标注模型。

7.根据权利要求1-5任一项所述的方法,其特征在于,在所述将所述目标轮廓作为所述待标注图像的标注数据之后,还包括:

针对所述待标注图像的标注数据发起人工审核请求;

接收对所述待标注图像的标注数据的人工确认信息或人工修改信息。

8.一种图像标注装置,其特征在于,包括:

模型预标注模块,用于根据目标标注模型对待标注图像进行预标注,生成掩膜图像;

标注数据调整模块,用于根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;

标注数据确认模块,用于将所述目标轮廓作为所述待标注图像的标注数据。

9.一种计算机设备,其特征在于,所述计算机设备包括:

一个或多个处理器;

存储器,用于存储一个或多个程序,

当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现如权利要求1-7中任一所述的图像标注方法。

10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现如权利要求1-7中任一所述的图像标注方法。

技术总结
本发明实施例公开了一种图像标注方法、装置、设备和介质。该方法包括:根据目标标注模型对待标注图像进行预标注,生成掩膜图像;根据所述掩膜图像选取所述待标注图像中目标对象的至少一个待定轮廓,并在所述至少一个待定轮廓中确定目标轮廓;将所述目标轮廓作为所述待标注图像的标注数据。上述技术方案,在提高标注效率的基础上,还提升了模型标注结果的准确率,以此也可降低人工对模型标注结果的修改量。

技术研发人员:胡文魁
受保护的技术使用者:广州虎牙科技有限公司
技术研发日:2020.01.13
技术公布日:2020.06.09

转载请注明原文地址: https://bbs.8miu.com/read-18513.html

最新回复(0)