[发明专利]训练数据的生成方法、装置、终端设备及存储介质

说明书

本申请实施例适用于人工智能技术领域，提供了一种训练数据的生成方法、装置、终端设备及存储介质，该方法包括：在虚拟环境中识别至少一个视角下的虚拟目标，生成虚拟图像；虚拟目标由虚拟环境对配置文件进行配置处理生成；根据配置文件的配置信息，获取虚拟图像的标注类别；基于虚拟图像和标注类别，生成虚拟目标的训练数据。采用上述方法可以在采集用于对真实目标进行训练的训练数据时，降低时间成本和人力成本。

技术领域

本申请属于人工智能技术领域，尤其涉及一种训练数据的生成方法、装置、终端设备及存储介质。

背景技术

目前，在进行人工智能模型识别的训练过程中，需要从真实环境中采集包含真实目标的图像，形成训练样本集进行模型训练。通常，该训练样本集需要包括真实目标在真实环境中的多种角度、形状或标签的图像。

目前，采用上述方式需要从真实环境中采集大量的图像，并通过人工对采集的图像进行标注，以得到训练数据。然而，在真实环境中，采集多种角度或形状下的真实目标的图像，通常需要实地考察，需要大量的时间成本和人力成本完成。

发明内容

本申请实施例提供了一种训练数据的生成方法、装置、终端设备及存储介质，可以解决在采集对真实目标进行训练的训练数据时，时间成本和人力成本均非常高的问题。

第一方面，本申请实施例提供了一种训练数据的生成方法，该方法包括：

在虚拟环境中识别至少一个视角下的虚拟目标，生成虚拟图像；虚拟目标由虚拟环境对配置文件进行配置处理生成；

根据配置文件的配置信息，获取虚拟图像的标注类别；

基于虚拟图像和标注类别，生成虚拟目标的训练数据。

第二方面，本申请实施例提供了一种训练数据的生成装置，该装置包括：

虚拟目标生成模块，用于在虚拟环境中识别至少一个视角下的虚拟目标，生成虚拟图像；虚拟目标由虚拟环境对配置文件进行配置处理生成；

获取模块，用于根据配置文件的配置信息，获取虚拟图像的标注类别；

训练数据生成模块，用于基于虚拟图像和标注类别，生成虚拟目标的训练数据。

第三方面，本申请实施例提供了一种终端设备，包括存储器、处理器以及存储在存储器中并可在处理器上运行的计算机程序，处理器执行计算机程序时实现如上述第一方面的方法。

第四方面，本申请实施例提供了一种计算机可读存储介质，计算机可读存储介质存储有计算机程序，计算机程序被处理器执行时实现如上述第一方面的方法。

第五方面，本申请实施例提供了一种计算机程序产品，当计算机程序产品在终端设备上运行时，使得终端设备执行上述第一方面的方法。

本申请实施例与现有技术相比存在的有益效果是：通过在虚拟环境中构建虚拟目标，以模拟真实目标在真实环境下的形状、大小、颜色等信息，并识别出该虚拟目标在不同视角下的虚拟图像，以代替真实目标在真实环境下被拍摄的图像进行处理。以此，终端设备可以降低工作人员去真实环境下采集该真实目标的图像所需花费的时间成本和人力成本。并且，根据配置信息中的标注类别，在虚拟图像具有多个时，终端设备自动化的为每个虚拟图像均进行标注，提升虚拟图像被标注的效率，以及训练数据的生成效率。

附图说明

为了更清楚地说明本申请实施例中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。

图1是本申请一实施例提供的一种训练数据的生成方法的实现流程图；