一种获取图像训练数据的方法和装置制造方法及图纸

本发明专利技术涉及一种获取图像训练数据的方法和装置，该方法包括：通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；控制所述摄像机对变更后的所述训练图像进行拍摄，以获取各所述最终3D虚拟场景下的图像训练数据。本方案提供了一种在虚拟场景中，获取可用于现实使用的训练数据的方法，提升了效率，降低了成本。降低了成本。降低了成本。

【技术实现步骤摘要】
一种获取图像训练数据的方法和装置


[0001]本专利技术涉及数据处理
，具体涉及一种获取图像训练数据的方法、装置、终端及存储介质。

技术介绍

[0002]目前图像引擎被大量应用在各种不同的领域，例如图像识别等，但是图像引擎的训练需要用到大量的数据，而目前用于图像引擎训练的数据，多采用人工采集与标注的方式来生成，但是目前的这种方式数据采集成本高，包括费用高，时间周期长，标注成本高等问题
[0003]由此，目前需要有一种更好的方法来解决现有技术中的问题。

技术实现思路

[0004]本专利技术提供一种获取图像训练数据的方法和装置，能够解决成本高，周期长的技术问题。
[0005]本专利技术解决上述技术问题的技术方案如下：
[0006]本专利技术实施例提出了一种获取图像训练数据的方法，包括：
[0007]通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；
[0008]通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；
[0009]控制所述摄像机对变更后的所述训练图像进行拍摄，以获取各所述最终3D虚拟场景下的图像训练数据。
[0010]在一个具体的实施例中，所述3D引擎包括：Unity3D引擎、虚幻引擎、寒霜引擎或起源引擎。
[0011]在一个具体的实施例中，所述通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景，包括：
[0012]根据预设的训练要求，设置训练脚本模块参数；
[0013]基于所述训练脚本模块参数在所述初始3D虚拟场景中对所述训练图像、所述载体、所述摄像机及所述光源中一个或多个进行变更的方式生成多个最终3D虚拟场景。
[0014]在一个具体的实施例中，所述训练脚本模块参数还用于设置图像导出命名规则及目录。
[0015]在一个具体的实施例中，所述训练图像的变更包括变更所述训练图像的内容；所述载体的变更包括变更所述载体的材质与颜色；所述摄像机的变更包括变更所述摄像机的数量与方位；所述光源的变更包括变更所述光源的数量与方位。
[0016]本专利技术实施例还提出了一种获取图像训练数据的装置，包括：
[0017]建立模块，用于通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练
图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；
[0018]变更模块，用于通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；
[0019]生成模块，用于控制所述摄像机对变更后的所述训练图像进行拍摄，以获取各所述最终3D虚拟场景下的图像训练数据。
[0020]在一个具体的实施例中，所述3D引擎包括：Unity3D引擎、虚幻引擎、寒霜引擎或起源引擎。
[0021]在一个具体的实施例中，所述变更模块，用于：
[0022]根据预设的训练要求，设置训练脚本模块参数；
[0023]基于所述训练脚本模块参数在所述初始3D虚拟场景中对所述载体、所述摄像机以及所述光源进行变更的方式生成多个3D虚拟场景。
[0024]在一个具体的实施例中，所述训练脚本模块参数还用于设置图像导出命名规则及目录。
[0025]在一个具体的实施例中，所述训练图像的变更包括变更所述训练图像的内容；所述载体的变更包括变更所述载体的材质与颜色；所述摄像机的变更包括变更所述摄像机的数量与方位；所述光源的变更包括变更所述光源的数量与方位。
[0026]本专利技术的有益效果是：
[0027]本专利技术实施例提出了一种获取图像训练数据的方法和装置，该方法包括：通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；控制所述摄像机对变更后的所述训练图像进行拍摄，以获取各所述最终3D虚拟场景下的图像训练数据。本方案提供了一种在虚拟场景中，获取可用于现实使用的训练数据的方法，提升了效率，降低了成本。
附图说明
[0028]图1为本专利技术实施例提供的一种获取图像训练数据的方法的框架流程示意图；
[0029]图2为本专利技术实施例提供的一种获取图像训练数据的方法的流程示意图；
[0030]图3为本专利技术实施例提供的一种获取图像训练数据的装置的框架结构示意图；
[0031]图4为本专利技术实施例提供的一种终端的框架结构示意图。
具体实施方式
[0032]以下结合附图对本专利技术的原理和特征进行描述，所举实例只用于解释本专利技术，并非用于限定本专利技术的范围。
[0033]实施例1
[0034]本专利技术实施例1提供的一种获取图像训练数据的方法，如图1与图2所示，包括以下步骤：
[0035]步骤101、通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；
[0036]具体的，如图2所示，在3D引擎中，根据图像数据需求，创建虚拟场景。场景中包括用于放置训练图像的支撑物(如桌子)，训练图像的载体(如纸张)，置于载体空间Y轴上方的摄像机位(可以有多个)，置于Y轴上方的光源(可以有多个)。
[0037]此外，所述3D引擎包括：Unity3D引擎、虚幻引擎、寒霜引擎或起源引擎。
[0038]除了上述列举的3D引擎外，还可以选取目前已有的其他3D引擎，只要能渲染实现生成不同的训练图像即可，并不限于上述的几种具体的3D引擎。
[0039]步骤102、通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；
[0040]进一步的，步骤102中的所述通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景，包括：
[0041]根据预设的训练要求，设置训练脚本模块参数；
[0042]基于所述训练脚本模块参数在所述初始3D虚拟场景中对所述训练图像、所述载体、所述摄像机及所述光源中一个或多个进行变更的方式生成多个最终3D虚拟场景。
[0043]具体的，根据训练要求，设定智能训练脚本模块参数(包括光源数量和方向，摄像机位数量和方向，图像载体材质及颜色，对训练图像进行变更，当然也可以对训练图像的内容进行变更。
[0044]具体的，所述训练图像的变更包括变更所述训练图像的内容；所述载体的变更包括变更所述载体本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种获取图像训练数据的方法，其特征在于，包括：通过3D引擎建立初始3D虚拟场景；所述初始3D虚拟场景包括：训练图像的载体、用于放置所述训练图像的支撑物、位于所述训练图像上方的摄像机以及光源；通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景；控制所述摄像机对变更后的所述训练图像进行拍摄，以获取各所述最终3D虚拟场景下的图像训练数据。2.如权利要求1所述的方法，其特征在于，所述3D引擎包括：Unity3D引擎、虚幻引擎、寒霜引擎或起源引擎。3.如权利要求1所述的方法，其特征在于，所述通过在所述初始3D虚拟场景中变更所述训练图像、所述载体、所述摄像机及所述光源中一个或多个的方式生成多个最终3D虚拟场景，包括：根据预设的训练要求，设置训练脚本模块参数；基于所述训练脚本模块参数在所述初始3D虚拟场景中对所述训练图像、所述载体、所述摄像机及所述光源中一个或多个进行变更的方式生成多个最终3D虚拟场景。4.如权利要求1所述的方法，其特征在于，所述训练脚本模块参数还用于设置图像导出命名规则及目录。5.如权利要求1或3所述的方法，其特征在于，所述训练图像的变更包括变更所述训练图像的内容；所述载体的变更包括变更所述载体的材质与颜色；所述摄像机的变更包括变更所述摄像机的数量与方位；所述...

【专利技术属性】
技术研发人员：周海，李彬，宋琪，康恒，
申请(专利权)人：云知声智能科技股份有限公司，
类型：发明
国别省市：