[发明专利]图像标注方法、图像标注装置及计算机存储介质

说明书

本发明涉及计算机视觉技术领域，提供一种图像标注方法、图像标注装置及计算机存储介质。图像标注方法，包括：获取待标注图像；基于待标注图像，确定分割区域的最小尺寸；基于最小尺寸，将待标注图像进行超像素分割，得到分割图像；基于当前分割图像确定当前阈值；根据当前阈值，采用区域融合的方式将分割图像的分割区域进行融合，得到当前融合图像；对局部区域中仅包括一个目标图像的局部区域进行标注；判断当前融合图像是否包括未标注的局部区域，根据判断结果完成待标注图像的标注。根据本公开提供的图像标注方法，能够有利于降低人工标注的时间成本，提高标注效率，且不影响图像标注的准确度。

技术领域

本发明一般地涉及计算机视觉技术领域，特别是涉及一种图像标注方法、图像标注装置及计算机存储介质。

背景技术

在计算机视觉领域中，越来越多的项目涉及到语义分割，例如：无人驾驶、视频理解等项目。随着语义分割标注的需求逐渐增多，标注的要求也越来越高，例如：全像素语义分割标注属于附加值极高的标注任务。

相关技术中，由于标注的精确度需要精确到像素点，采用磁力贴合线等方法；或者，采用Photoshop等工具通过在边缘打点的方式勾勒出待分割物体的边缘进行人工标注。采用磁力贴合线等方法容易存在较大误差，采用人工标注方式则非常耗时，最终结果的精确度与人工打点数量成正比，标注精度往往无法达到项目的精度需求。

发明内容

为了解决现有技术中存在的上述问题，本发明提供一种图像标注方法、图像标注装置及计算机存储介质。

第一方面，本发明实施例提供一种图像标注方法，包括：获取待标注图像，其中待标注图像中包括一个或多个目标图像；基于待标注图像，确定分割区域的最小尺寸；基于最小尺寸，将待标注图像进行超像素分割，得到分割图像；基于当前分割图像确定当前阈值；根据当前阈值，采用区域融合的方式将分割图像的分割区域进行融合，得到当前融合图像，融合图像包括一个或多个局部区域；对局部区域中仅包括一个目标图像的局部区域进行标注；判断当前融合图像是否包括未标注的局部区域，若当前融合图像包括未标注的局部区域，则将未标注的局部区域作为新的当前分割图像，并返回基于当前分割图像确定当前阈值的步骤；若当前融合图像不包括未标注的局部区域，则完成待标注图像的标注。

在一实施例中，根据当前阈值，采用区域融合的方式将分割图像的分割区域进行融合，得到当前融合图像，包括：获取分割图像中各个分割区域内的像素的第一平均色彩值；根据当前阈值和第一平均色彩值，采用区域融合的方式，将分割图像中两个第一平均色彩值的差小于当前阈值的分割图像中对应的两个分割区域进行融合，得到当前融合图像。

在一实施例中，基于当前分割图像确定当前阈值，包括：获取当前分割图像中未标注的区域的像素的第二平均色彩值；基于第二平均色彩值，确定当前阈值。

在另一实施例中，对局部区域中仅包括一个目标图像的局部区域进行标注，包括：根据目标图像的类别，对局部区域中仅包括一个目标图像的局部区域进行标注。

在一实施例中，基于待标注图像，确定分割区域的最小尺寸，包括：通过目标检测模型，确定待标注图像中的目标图像的坐标信息；基于坐标信息，确定分割区域的最小尺寸。

在另一实施例中，图像标注方法还包括：通过目标检测模型，确定待标注图像中的目标图像的类别信息；基于坐标信息和类别信息，确定目标图像的中心坐标，其中，中心坐标与类别信息对应；根据目标图像的类别，对局部区域中仅包括一个目标图像的局部区域进行标注，包括：若局部区域中仅包括一个目标图像的中心坐标，则根据中心坐标对应的类别信息，将局部区域进行标注。