[发明专利]基于3D游戏的数据集构建方法、装置、设备及存储介质

权利要求书

1.一种基于3D游戏的数据集构建方法,其特征在于，所述方法包括：

获取目标3D游戏的编译后的游戏数据抓取脚本；

运行所述目标3D游戏，根据所述游戏数据抓取脚本得到游戏画面数据，所述游戏画面数据包括图像数据和非图像数据；

对所述游戏画面数据进行数据清洗得到待标注的样本数据；

构建数据标注系统，通过所述数据标注系统对所述样本数据进行标注。

2.根据权利要求1所述的方法，其特征在于，所述构建数据标注系统，通过所述数据标注系统对所述样本数据进行标注，包括：

构建基于移动终端的数据标注系统，所述数据标注系统包括用户登录组件和数据标注组件；

所述用户登录组件用于使用户通过在用户登录页面输入账号与密码登录所述数据标注系统；

所述数据标注组件用于对数据标注页面内展示的样本数据进行打分与描述。

3.根据权利要求1所述的方法，其特征在于，在对所述样本数据进行标注之后，所述方法还包括：

通过标注后的样本数据对单目深度图估计模型、无绿幕抠图模型、文本生成布局以及人体骨骼关键点预测模型进行训练。

4.根据权利要求1所述的方法，其特征在于，所述游戏画面数据包括游戏截图、深度图以及模板图；所述非图像数据包括摄像机参数信息、物体2D包围盒数据、物体3D包围盒数据、物体3D模型名称以及人体3D骨骼关键点信息。

5.根据权利要求4所述的方法，其特征在于，所述游戏画面数据还包括：

基于所述模板图通过膨胀腐蚀法来得到当前帧的三元图，所述膨胀腐蚀法具体包括使用OpenCV中的cv2.dilate和cv2.erode。

6.根据权利要求1所述的方法，其特征在于，所述根据所述游戏数据抓取脚本得到游戏画面数据，包括：

通过人为判断当前游戏页面质量满足要求时，根据所述游戏数据抓取脚本自动捕获当前游戏画面数据；其中，所述游戏页面质量包括画面中实体的数量、画面中实体的相对位置、画面中人物的行为动作。

7.根据权利要求1所述的方法，所述目标3D游戏为GTAV，其特征在于，所述根据所述游戏数据抓取脚本得到游戏画面数据，包括：

使用GTAVisionExport工具来从缓存中获取图像数据；

基于ScriptHookVDotNet游戏脚本库来从GTAV游戏中获取非图像数据。

8.一种基于3D游戏的数据集构建装置，其特征在于，所述装置包括：

编译模块，用于获取目标3D游戏的编译后的游戏数据抓取脚本；

获取模块，用于运行所述目标3D游戏，根据所述游戏数据抓取脚本得到游戏画面数据，所述游戏画面数据包括图像数据和非图像数据；

预处理模块，用于对所述游戏画面数据进行数据清洗得到待标注的样本数据；

标注模块，用于构建数据标注系统，通过所述数据标注系统对所述样本数据进行标注。

9.一种设备，其特征在于，包括存储器和处理器，所述存储器存储有计算机程序，所述计算机程序被所述处理器执行时实现如权利要求1至7任一所述的基于3D游戏的数据集构建方法。

10.一种计算机可读存储介质，其特征在于，其上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至7任一所述的基于3D游戏的数据集构建方法。