[发明专利]图像标注方法及终端设备

权利要求书

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

获取待标注图像，并将所述待标注图像输入目标检测模型得到所述待标注图像的第一标注集合；

显示包含所述第一标注集合的待标注图像；

获取用户输入的调整信息，并根据所述调整信息对所述第一标注集合进行调整；

根据调整后的第一标注集合确定所述待标注图像的最终标注集合。

2.如权利要求1所述的图像标注方法，其特征在于，所述将所述待图像输入目标检测模型得到所述待标注图像的第一标注集合包括：

获取原始图像，并对所述原始图像进行数据增强；

建立所述目标检测模型，并根据数据增强后的原始图像对所述目标检测模型进行训练；

将所述待标注图像输入训练后的目标检测模型得到所述第一标注集合。

3.如权利要求2所述的图像标注方法，其特征在于，所述获取原始图像，并对所述原始图像进行数据增强包括：

获取预设张数的原始图像及对应的人工标注信息；所述人工标注信息为对原始图像的目标区域进行人工标注得到的信息；

对预设张数的原始图像及对应的人工标注信息进行数据增强；

所述根据数据增强后的原始图像对所述目标检测模型进行训练包括：

根据数据增强后的原始图像及对应的人工标注信息对所述目标检测模型进行训练。

4.如权利要求1所述的图像标注方法，其特征在于，所述第一标注集合中的每个标注元素的信息包括标注框两个相对顶点的坐标、标注类型和置信度；

所述显示包含所述第一标注集合的待标注图像包括：

根据所述第一标注集合中各个标注元素的信息对所述待标注图像进行绘制处理；所述绘制处理包括根据各个标注元素中标注框两个相对顶点的坐标，在所述待标注图像上绘制各个标注元素对应的标注框，并在各个标注框的预设位置显示各个标注元素的标注类型和置信度；

显示经过所述绘制处理后的待标注图像。

5.如权利要求4所述的图像标注方法，其特征在于，所述调整信息包括标注框两个相对顶点的调整坐标；

所述根据所述调整信息对所述第一标注集合进行调整包括：

根据标注框两个相对顶点的调整坐标更新所述第一标注集合中对应的标注元素的信息。

6.如权利要求1至5任一项所述的图像标注方法，其特征在于，在所述显示包含所述第一标注集合的待标注图像之后，所述根据调整后的第一标注集合确定所述待标注图像的最终标注集合之前，还包括：

获取用户输入的新增标注信息，并根据所述新增标注信息生成第二标注集合；

所述根据调整后的第一标注集合确定所述待标注图像的最终标注集合包括：

根据调整后的第一标注集合和所述第二标注集合确定所述待标注图像的最终标注集合。

7.一种图像标注装置，其特征在于，包括：

获取模块，用于获取待标注图像，并将所述待标注图像输入目标检测模型得到所述待标注图像的第一标注集合；

显示模块，用于显示包含所述第一标注集合的待标注图像；

调整模块，用于获取用户输入的调整信息，并根据所述调整信息对所述第一标注集合进行调整；

处理模块，用于根据调整后的第一标注集合确定所述待标注图像的最终标注集合。

8.如权利要求7所述的图像标注装置，其特征在于，所述获取模块用于：

获取原始图像，并对所述原始图像进行数据增强；

建立所述目标检测模型，并根据数据增强后的原始图像对所述目标检测模型进行训练；

将所述待标注图像输入训练后的目标检测模型得到所述第一标注集合。

9.一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如权利要求1至6任一项所述方法的步骤。

10.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述方法的步骤。