一种三维位姿的构建方法、设备及存储介质技术

本公开关于一种三维位姿的构建方法、设备及存储介质，涉及图像处理领域。本公开实施例至少解决相关技术中，预测建筑的三维位姿的效果鲁棒性较差的问题。该方法包括：获取待构建图像的图像数据，待构建图像包括待构建建筑；将待构建图像的图像数据输入到预先训练好的分割模型，预测得到目标分割数据以及目标法向数据；目标分割数据包括至少一个平面上的像素点的二维坐标，至少一个平面为在待构建图像上待构建建筑的平面，目标法向数据包括待构建图像上的像素点对应的法向量；基于目标分割数据以及目标法向数据，确定至少一个平面上的像素点在三维坐标系中对应的三维坐标，以得到待构建建筑的三维位姿。建建筑的三维位姿。建建筑的三维位姿。

【技术实现步骤摘要】
一种三维位姿的构建方法、设备及存储介质


[0001]本公开涉及计算机领域，尤其涉及一种三维位姿的构建方法、设备以及存储介质。

技术介绍

[0002]随着科技的不断发展和互联网的普及，在一些虚拟现实场景任务中，可以构建二维图像中建筑的三维位姿。具体的，电子设备在获取包括建筑的二维图像之后，获取建筑在二维图像上的角点的二维坐标，进而根据训练好的热图(heatmap)模型，以及二维图像在三维坐标系中的投影关系，预测二维图像中每个角点坐标在三维坐标系中对应的三维坐标，这样，即可以确定建筑的三维位姿。
[0003]但是，在上述构建三维位姿的过程中，由于仅通过二维图像上可以识别到的角点对建筑进行三维位姿的预测，很大程度上只能够反映建筑在三维坐标系中的轮廓，这就导致建筑的三维位姿的预测效果不连续，鲁棒性较差。

技术实现思路

[0004]本公开提供一种三维位姿的构建方法、设备及存储介质，以至少解决相关技术中，预测建筑的三维位姿的效果鲁棒性较差的问题。本公开的技术方案如下：
[0005]根据本公开实施例的第一方面，提供一种三维位姿本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种三维位姿的构建方法，其特征在于，包括：获取待构建图像的图像数据，所述待构建图像包括待构建建筑；将所述待构建图像的图像数据输入到预先训练好的分割模型，预测得到目标分割数据以及目标法向数据；所述目标分割数据包括至少一个平面上的像素点的二维坐标，所述至少一个平面为在所述待构建图像上所述待构建建筑的平面，所述目标法向数据包括所述待构建图像上的像素点对应的法向量；基于所述目标分割数据以及所述目标法向数据，确定所述至少一个平面上的像素点在三维坐标系中对应的三维坐标，以得到所述待构建建筑的三维位姿。2.根据权利要求1所述的三维位姿的构建方法，其特征在于，所述方法还包括：将获取到的训练样本输入初始分割模型，以训练得到所述分割模型；所述训练样本包括多个样本图像的图像数据，以及每个样本图像的样本分割数据和样本法向数据；所述样本分割数据用于标识所述每个样本图像所包括的建筑的平面上的样本像素点，所述样本法向数据用于标识所述样本像素点在对应平面上的法向量。3.根据权利要求2所述的三维位姿的构建方法，其特征在于，所述方法还包括：获取所述每个样本图像中建筑的角点在所述每个样本图像上对应的二维坐标；基于所述每个样本图像中建筑的角点对应的二维坐标以及预设的投影关系，确定所述每个样本图像中建筑的角点在所述三维坐标系中对应的三维坐标；所述投影关系包括所述每个样本图像所在的二维坐标系与所述三维坐标系之间的投影关系；基于所述每个样本图像中建筑的角点在所述三维坐标系中对应的三维坐标，确定所述每个样本图像中建筑的平面法向量，并基于确定到所述每个样本图像中建筑的平面法向量，确定所述每个样本图像对应的样本法向数据；所述每个样本图像中建筑的平面法向量为所述每个样本图像中的建筑在所述每个样本图像上的至少一个平面的法向量。4.根据权利要求2所述的三维位姿的构建方法，其特征在于，所述将获取到的训练样本输入初始分割模型，以训练得到所述分割模型，包括：将所述每个样本图像的图像数据输入所述初始分割模型中，以分别确定所述每个样本图像的分割损失以及法向损失；所述分割损失用于反映所述每个样本图像的预测分割数据与所述每个样本图像的所述样本分割数据之间的误差，所述法向损失用于反映所述每个样本图像的预测法向数据与所述每个样本图像的所述样本法向数据之间的误差；基于确定到的所述分割损失及所述法向损失，更新得到所述初始分割模型，以训练得到所述分割模型。5.根据权利...

【专利技术属性】
技术研发人员：蔡东阳，王再冉，郭小燕，
申请(专利权)人：北京达佳互联信息技术有限公司，
类型：发明
国别省市：