[发明专利]三维模型构建方法、装置、计算机设备及存储介质

说明书

本申请涉及一种三维模型构建方法、装置、计算机设备和存储介质。该方法包括：获取二维图像，二维图像包括待构建三维模型的目标对象；基于二维图像获取目标对象的多个映射点的三维坐标和各映射点的顶点相似度预测值，各映射点的顶点相似度预测值表示映射点与距离映射点最近的目标对象的三维顶点的相似度；基于各映射点的顶点相似度预测值，从多个映射点中筛选出多个目标映射点；基于多个目标映射点的三维坐标构建目标对象的三维模型。本申请实施例中，基于各映射点的顶点相似度预测值对各映射点进行筛选处理可以得到过滤掉一些不属于顶点的映射点，从而更好地适应不同顶点数量的目标对象的三维模型重建。

技术领域

本申请涉及计算机视觉技术领域，特别是涉及一种三维模型构建方法、装置、计算机设备及存储介质。

背景技术

近年来，随着智能机器人、无人驾驶技术、虚拟现实技术等快速发展，基于二维图像的三维模型重建是计算机视觉领域的一个非常重要的研究方向。

现有技术中，基于二维图像进行三维模型重建的过程一般是：将二维图像输入到训练好的神经网络模型中，得到神经网络模型输出的映射点的三维坐标，其中，映射点的数量是在训练神经网络模型的过程中预先设定好的。

然而，上述方法中，在对顶点数量较少的目标对象进行三维模型重构时，重构的目标对象的三维模型的顶点上会出现过多的映射点，导致模型糊化，因此，三维模型的重构精度不好。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高目标对象的三维模型的精度的三维模型构建方法、装置、计算机设备及存储介质。

一种三维模型构建方法，该方法包括：

获取二维图像，二维图像包括待构建三维模型的目标对象；

基于二维图像获取目标对象的多个映射点的三维坐标和各映射点的顶点相似度预测值，各映射点的顶点相似度预测值表示映射点与距离映射点最近的目标对象的三维顶点的相似度；

基于各映射点的顶点相似度预测值，从多个映射点中筛选出多个目标映射点；

基于多个目标映射点的三维坐标构建目标对象的三维模型。

在其中一个实施例中，基于各映射点的顶点相似度预测值，从多个映射点中筛选出多个目标映射点，包括：

基于各映射点的顶点相似度预测值对各映射点进行降序排列，并按照降序排列顺序依次将各映射点确定为候选映射点；

计算其他映射点到候选映射点的距离，剔除到候选映射点的距离小于距离阈值的映射点，得到多个目标映射点。

在其中一个实施例中，剔除到候选映射点的距离小于距离阈值的映射点之后，方法还包括：

从多个映射点中剔除顶点相似度预测值小于相似度阈值的映射点。

在其中一个实施例中，基于各映射点的顶点相似度预测值，从多个映射点中筛选出多个目标映射点之前，方法还包括：

对各映射点的三维坐标进行去中心化处理，得到去中心化处理后的多个第一映射点；

对各第一映射点进行归一化处理，得到多个第二映射点；

基于各映射点的顶点相似度预测值，从多个映射点中筛选出多个目标映射点，包括：

基于各第二映射点的顶点相似度预测值，从多个第二映射点中筛选出多个目标映射点。

在其中一个实施例中，对各映射点的三维坐标进行去中心化处理，得到去中心化处理后的多个第一映射点，包括：

获取多个映射点的三维坐标中每一维坐标轴的坐标平均值；