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

权利要求书

1.一种数据生成方法，包括：

获取第一图像数据，其中，所述第一图像数据为表示用户所处真实环境的数据；

获取目标对象的类别信息和平面信息，其中，所述目标对象为所述第一图像数据中的对象，所述类别信息为表示所述目标对象的物体类型的信息，所述平面信息包括所述目标对象的外表面的信息；

获取第二图像数据，其中，所述第二图像数据为包含虚拟对象的数据；

根据所述类别信息和所述平面信息，混合所述第一图像数据和所述第二图像数据，生成目标图像数据，其中，所述目标图像数据为包含所述目标对象和所述虚拟对象的数据，

其中，所述根据所述平面信息和所述类别信息，混合所述第一图像数据和所述第二图像数据，生成目标图像数据，包括：

根据所述类别信息，确定所述第二图像数据中的所述虚拟对象与所述第一图像数据中的所述目标对象之间的相对位置关系；

根据所述平面信息和所述相对位置关系，将所述虚拟对象渲染至所述目标对象的预设位置处，获得所述目标图像数据。

2.根据权利要求1所述的方法，所述获取所述目标对象的类别信息和平面信息，包括：

将所述第一图像数据输入到目标图像分割模型中，获得所述目标对象的掩膜信息；

根据所述掩膜信息，获得所述类别信息和所述平面信息。

3.根据权利要求2所述的方法，所述根据所述掩膜信息，获得所述类别信息，包括：

将所述掩膜信息输入到目标类别识别模型中，获得所述类别信息。

4.根据权利要求2所述的方法，所述根据所述掩膜信息，获得所述平面信息，包括：

根据所述掩膜信息，获得所述目标对象在所述第一图像数据中对应的目标图像块；

根据所述目标图像块，获取所述目标对象的关键点在世界坐标系下的目标位置信息，其中，所述关键点包括所述目标对象的角点；

根据所述目标位置信息和预设平面拟合算法，获得所述平面信息，其中，所述平面信息包括与所述目标对象的每一平面对应的中心点坐标和平面法向量。

5.根据权利要求4所述的方法，所述方法应用于电子设备，所述根据所述目标图像块，获取所述目标对象的关键点在世界坐标系下的目标位置信息，包括：

根据所述目标图像块，检测所述关键点在所述第一图像数据中的第一位置信息；

获取所述电子设备在第一时刻的位姿信息，以及，所述关键点在第二时刻获取到的第三图像数据中的第二位置信息，其中，所述第一时刻包括当前时刻，所述第二时刻早于所述第一时刻；

根据所述第一位置信息、所述位姿信息和所述第二位置信息，获得所述目标位置信息。

6.根据权利要求3所述的方法，所述目标图像分割模型和所述目标类别识别模型通过以下步骤训练获得：

获取样本数据，其中，所述样本数据为包含预设场景中的样本对象的数据；

根据所述样本数据，联合训练初始图像分割模型和初始类别识别模型，获得所述目标图像分割模型和所述目标类别识别模型。

7.根据权利要求1所述的方法，在获得所述目标图像数据之后，所述方法还包括：

展示所述目标图像数据。