[发明专利]三维模型检索方法、装置、电子设备及存储介质

说明书

本申请公开了一种三维模型检索方法、装置、电子设备及存储介质，其中，方法包括：获取目标三维模型的残缺点云数据；提取残缺点云数据的点云特征；逐一计算点云特征与预设数据库中每个三维模型的点云特征的欧式距离，并将欧式距离最小的三维模型作为目标三维模型的检索结果。由此，解决了相关技术中采集的点云数据残缺，导致检索效果较差，且需要大量成对标注数据支撑，使得可用的标注样本规模受限，检索性能及效率较低等技术问题。

技术领域

本申请涉及计算机视觉及模型预训练技术领域，特别涉及一种三维模型检索方法、装置、电子设备及存储介质。

背景技术

三维模型检索技术能够在大规模数据中寻找相似的三维点云模型，进而对物体级别的点云数据进行细粒度的分析，被广泛应用于自动驾驶、缺陷检测等工业领域。

相关技术采用实体级别的三维模型检索技术，可以改善类型级别的检索难以应用于自动驾驶、智能装备等精细化场景的问题。

相关技术要求检索结果与输入模型属于同一个实体，因此，相关技术需要获取三维场景中的点云数据，而三维场景中的点云数据通常由扫描类传感器获取，如激光雷达，其数据获取方式在真实的采集环境下，被观测的物体往往因为各类遮挡，导致采集到的点云数据是残缺的，无法获知被观测物体的完整形状，存在局限性。

此外，由于深度学习具有数据驱动的特性，相关技术中的实体级别的三维模型检索任务需要大量的成对标注数据的支撑，且相比于二维图像数据，三维点云数据更加复杂，这导致了三维模型检索技术可用的标注样本规模非常受限，且代价巨大。

综上，相关技术中难以避免采集到的点云数据残缺，无法获得被感知物体的完整形状，且需要大量成对标注数据支撑，使得可用的标注样本规模受限，难以应用于复杂场景，有待改善。

发明内容

本申请提供一种三维模型检索方法、装置、电子设备及存储介质，以解决相关技术中采集的点云数据残缺，导致检索效果较差，且需要大量成对标注数据支撑，使得可用的标注样本规模受限，检索性能及效率较低等技术问题。

本申请第一方面实施例提供一种三维模型检索方法，包括以下步骤：获取目标三维模型的残缺点云数据；提取所述残缺点云数据的点云特征；以及逐一计算所述点云特征与预设数据库中每个三维模型的点云特征的欧式距离，并将所述欧式距离最小的三维模型作为所述目标三维模型的检索结果。

可选地，在本申请的一个实施例中，所述逐一计算所述点云特征与预设数据库中每个三维模型的点云特征的欧式距离，包括：对每个点云特征进行分段，并在分段后的子空间中进行聚类，生成所述每个三维模型的特征编号；计算各子空间中簇中心点间的距离，以得到所述欧式距离。

可选地，在本申请的一个实施例中，在逐一计算所述点云特征与所述预设数据库中每个三维模型的点云特征的欧式距离之前，还包括：获取所述每个三维模型的体素集合，并生成所述体素集合的体素掩码，以移除被遮蔽的体素块中的点云，得到对应的呈块状缺失的残缺点云模型；基于所述呈块状缺失的残缺点云模型，通过非对称的编解码结构补全对应模型，并在预训练结束后，得到点云编码器的网络权重，以提取所述点云特征；基于所述点云编码器的网络权重，构建实体间三元组和视角间三元组。

可选地，在本申请的一个实施例中，所述获取所述每个三维模型的体素集合，并生成所述体素集合的体素掩码，以移除被遮蔽的体素块中的点云，得到对应的呈块状缺失的残缺点云模型，包括：利用立方体网格切割点云将所述每个三维模型体素化，得到所述体素集合；基于预设遮蔽率，随机生成所述每个体素的体素掩码，得到所述呈块状缺失的残缺点云模型。

可选地，在本申请的一个实施例中，所述基于所述呈块状缺失的残缺点云模型，通过非对称的编解码结构补全对应模型，并在预训练结束后，得到点云编码器的网络权重，包括：利用预设点云特征编码器对每个残缺点云模型进行特征提取，并利用预设点云特征解码器进行残缺点云模型的补全，得到补全结果；基于所述补全结果，利用倒角距离损失函数进行优化，得到所述点云编码器的网络权重。