[发明专利]一种图像数据集标注方法、装置和电子设备

说明书

本发明公开了一种图像数据集标注方法、装置和电子设备，属于自动驾驶技术领域。其中，方法包括：获取标注任务需求信息和待标注图像集，所述待标注图像集包括至少两帧待标注图像，所述标注任务需求信息包括场景需求信息和标注对象需求信息；通过预设模型对所述待标注图像集中的每一个待标注图像进行图像识别处理，以确定所述所述待标注图像集中的每一个待标注图像的预标注结果，然后，通过对目标待标注图像的预标注结果的第一输入，以更新得到所述目标待标注图像的目标标注结果；最后，通过对所述第一图像集中的全部待标注图像的目标标注结果进行纠错检测，并输出纠错结果。本发明能够提升图像标注过程的效率。

技术领域

本发明属于自动驾驶技术领域，具体涉及一种图像数据集标注方法、装置和电子设备。

背景技术

现有技术中，在智慧交通自动驾驶领域，基于路侧视觉感知的交通人工智能(Artificial Intelligence，AI)应用研究中，需要大量标注数据用于AI模型的训练，提高识别和检测精度，以支撑智慧交通自动驾驶应用领域的具体需求。而在标注过程中，往往采用的方式是人工标注的方式对大量数据进行标注，其标注效率低。

发明内容

本发明的目的是提供一种图像数据集标注方法、装置和电子设备，能够解决相关技术中的数据标注过程存在的标注效率低的问题。

为了解决上述技术问题，本发明是这样实现的：

第一方面，本发明提供了一种图像数据集标注方法，该方法包括：

获取标注任务需求信息和待标注图像集，其中，所述待标注图像集包括至少两帧待标注图像，所述标注任务需求信息包括场景需求信息和标注对象需求信息；

根据所述场景需求信息，采用预设模型对所述待标注图像集中的每一个待标注图像进行图像识别处理，以确定所述所述待标注图像集中的每一个待标注图像的预标注结果；

接收对目标待标注图像的预标注结果的第一输入，其中，所述目标待标注图像为第一图像集内的任一图像，第一图像集为所述待标注图像集中与所述标注对象需求信息和所述场景需求信息匹配的图像集合；

响应于所述第一输入，对所述目标待标注图像的预标注结果进行更新，以得到所述目标待标注图像的目标标注结果；

对所述第一图像集中的全部待标注图像的目标标注结果进行纠错检测，并输出纠错结果。

第二方面，本发明还提供了一种图像数据集标注装置，包括：

第一获取模块，用于获取标注任务需求信息和待标注图像集，其中，所述待标注图像集包括至少两帧待标注图像，所述标注任务需求信息包括场景需求信息和标注对象需求信息；

第一确定模块，用于根据所述场景需求信息，采用预设模型对所述待标注图像集中的每一个待标注图像进行图像识别处理，以确定所述所述待标注图像集中的每一个待标注图像的预标注结果；

第一接收模块，用于接收用户对目标待标注图像的预标注结果的第一输入，其中，所述目标待标注图像为第一图像集内的任一图像，第一图像集为所述待标注图像集中与所述标注对象需求信息和所述场景需求信息匹配的图像集合；

更新模块，用于响应于所述第一输入，对所述目标待标注图像的预标注结果进行更新，以得到所述目标待标注图像的目标标注结果；

纠错模块，用于对所述第一图像集中的全部待标注图像的目标标注结果进行纠错检测，并输出纠错结果。

第三方面，本发明还提供了一种电子设备，该电子设备包括处理器、存储器及存储在所述存储器上并可在所述处理器上运行的程序或指令，所述程序或指令被所述处理器执行时实现如第一方面所述的方法的步骤。