[发明专利]数据集生成方法、装置、电子设备及存储介质

权利要求书

1.一种数据集生成方法，其特征在于，包括：

确定预先构建的三维虚拟场景中目标对象的预标注信息；

根据模型训练任务的需求和所述预标注信息，生成数据集生成配置；

根据所述数据集生成配置，渲染生成基于所述三维虚拟场景的渲染图像，并生成所述渲染图像对应的标注文件；根据所述渲染图像和所述标注文件，生成对应于所述模型训练任务的数据集。

2.根据权利要求1所述的方法，其特征在于，根据模型训练任务的需求和所述预标注信息，生成数据集生成配置，包括：

根据通过预定的语言编写的定义指令，或响应于对待选的标注数据结构定义的选择指令，生成与所述模型训练任务对应的标注数据结构定义；

解析所述标注数据结构定义以生成标注数据结构框架，并根据所述预标注信息，为所述标注数据结构框架中的各属性进行赋值；

根据所述标注数据结构定义和赋值后的标注数据结构框架，生成所述数据集生成配置。

3.根据权利要求2所述的方法，其特征在于，根据所述预标注信息，为所述标注数据结构框架中的各属性进行赋值，包括：

确定所述标注数据结构框架中的目标属性；

获取根据所述预标注信息生成的与所述目标属性对应的脚本，根据所述脚本为所述目标属性进行赋值。

4.根据权利要求2所述的方法，其特征在于，根据所述标注数据结构定义和赋值后的标注数据结构框架，生成所述数据集生成配置，包括：

获取预先定义的渲染偏好信息和数据集存储信息；其中，所述渲染偏好信息用于设置渲染生成所述三维虚拟场景的相关参数；所述数据集存储信息用于设置数据集的存储方式和存储位置；

根据所述标注数据结构定义、赋值后的标注数据结构框架、所述渲染偏好信息和所述数据集存储信息，生成所述数据集生成配置。

5.根据权利要求4所述的方法，其特征在于，根据所述数据集生成配置，渲染生成基于所述三维虚拟场景的渲染图像，并生成所述渲染图像对应的标注文件，包括：

根据所述标注数据结构定义和赋值后的标注数据结构框架，生成所述渲染图像对应的标注文件；

根据所述数据集存储信息，将所述标注文件进行存储。

6.根据权利要求4所述的方法，其特征在于，根据所述数据集生成配置，渲染生成基于所述三维虚拟场景的渲染图像，并生成所述渲染图像对应的标注文件，包括：

根据所述渲染偏好信息，确定渲染生成所述三维虚拟场景的相关参数，根据所述相关参数请求相应的渲染服务生成所述三维虚拟场景，并根据所述三维虚拟场景中目标帧范围内包括的各帧图像，生成所述渲染图像；

根据所述数据集存储信息，将所述渲染图像进行存储。

7.根据权利要求1所述的方法，其特征在于，确定预先构建的三维虚拟场景中目标对象的预标注信息，包括：

确定所述三维虚拟场景中的目标对象的相关信息、目标渲染相机的相关信息和目标帧的相关信息；

确定所述目标对象表面各顶点的世界坐标，得到世界坐标集合；

将所述世界坐标转换为所述目标渲染相机画面中的二维坐标，得到二维坐标集合；

根据所述二维坐标集合，得到所述目标对象的边界顶点坐标集合；

根据所述边界顶点坐标集合，得到所述目标对象的检测框信息；

根据所述目标对象的相关信息、所述目标渲染相机的相关信息、所述目标帧的相关信息、所述世界坐标集合、所述二维坐标集合、所述边界顶点坐标集合和所述检测框信息，生成所述预标注信息。

8.一种数据集生成装置，其特征在于，包括：

确定模块，被配置为确定预先构建的三维虚拟场景中目标对象的预标注信息；

配置模块，被配置为根据模型训练任务的需求和所述预标注信息，生成数据集生成配置；

生成模块，被配置为根据所述数据集生成配置，渲染生成基于所述三维虚拟场景的渲染图像，并生成所述渲染图像对应的标注文件；根据所述渲染图像和所述标注文件，生成对应于所述模型训练任务的数据集。