[发明专利]生成训练样本的方法和装置

权利要求书

1.一种生成训练样本的方法，其特征在于，所述方法包括：

获取目标三维场景的场景信息；

基于所述场景信息，确定所述目标三维场景的虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置；

基于所述虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，确定训练样本。

2.根据权利要求1所述的方法，其特征在于，所述场景信息包括所述目标三维场景中至少一个物体的尺寸信息、位姿信息、属性信息以及光照信息。

3.根据权利要求2所述的方法，其特征在于，所述基于所述场景信息，确定所述目标三维场景的虚拟拍摄图像和所述虚拟拍摄图像中每个物体的图像位置，包括：

获取所述目标三维场景中虚拟相机的拍摄参数信息；

基于所述场景信息和所述拍摄参数信息，确定所述目标三维场景的虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置。

4.根据权利要求3所述的方法，其特征在于，所述基于所述场景信息和所述拍摄参数信息，确定所述目标三维场景的虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，包括：

基于所述尺寸信息，构建每个物体的三维模型；

基于每个物体的三维模型和所述位姿信息，生成所述目标三维场景对应的虚拟三维场景模型；

基于所述属性信息、所述光照信息、所述虚拟相机的拍摄参数信息和所述虚拟三维场景模型，确定每个物体的图像位置和颜色信息，其中，所述每个物体的图像位置包括所述每个物体的各像素点的图像位置；

基于所述每个物体的图像位置，确定所述虚拟拍摄图像中目标物体的图像位置；

基于所述每个物体的图像位置和颜色信息，确定所述目标三维场景模型的虚拟拍摄图像。

5.根据权利要求3所述的方法，其特征在于，所述拍摄参数信息包括所述虚拟相机的类型和视场角。

6.根据权利要求2所述的方法，其特征在于，所述光照信息包括光源的位置、强度和颜色。

7.根据权利要求2所述的方法，其特征在于，所述属性信息包括物体的材质、纹理和颜色。

8.根据权利要求1所述的方法，其特征在于，所述基于所述虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，确定训练样本，包括：

基于所述虚拟拍摄图像中目标物体的图像位置，生成所述虚拟拍摄图像对应的语义标签分割图，其中，所述目标物体的图像位置包括所述目标物体各像素点的图像位置；

将所述虚拟拍摄图像和所述语义标签分割图，确定为训练样本。

9.根据权利要求8所述的方法，其特征在于，所述基于所述虚拟拍摄图像中目标物体的图像位置，生成所述虚拟拍摄图像对应的语义标签分割图，包括：

确定所述目标物体对应的颜色信息；

基于所述目标物体的图像位置和所述目标物体对应的颜色信息，生成所述虚拟拍摄图像对应的语义标签分割图。

10.根据权利要求9所述的方法，其特征在于，所述目标物体为所述虚拟拍摄图像中的所有物体。

11.根据权利要求1所述的方法，其特征在于，所述基于所述虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，确定训练样本，包括：

将所述虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，确定为训练样本。

12.一种生成训练样本的装置，其特征在于，所述装置包括：

获取模块，用于获取目标三维场景的场景信息；

生成模块，用于基于所述场景信息，确定所述目标三维场景的虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置；

确定模块，用于基于所述虚拟拍摄图像和所述虚拟拍摄图像中目标物体的图像位置，确定训练样本。