一种基于锚点检测的点云补全方法技术

本发明专利技术属于点云补全技术领域，涉及一种基于锚点检测的点云补全方法，该方法首先使用局部删除操作对数据集做预处理，从而模拟不完整的点云；然后将点云所在空间划分为多个锚点，训练神经网络判断锚点的正负性，得到密度均匀，形状规整的正样本锚点；最后在正样本锚点的邻域内通过神经网络生成密集点云，实现由单一锚点向局部的扩散，完成点云的补全。为实现该方法，我们训练了与之适配的GCNet模型，该神经网络模型能够判断锚点的正负性以及生成锚点扩散的三维坐标信息，实现了端到端的点云补全，并解决了现有方法只依靠神经网络回归生成点云坐标而导致的补全效果不佳的问题，提高了点云补全的效果。点云补全的效果。点云补全的效果。

【技术实现步骤摘要】
一种基于锚点检测的点云补全方法


[0001]本专利技术属于点云补全领域，具体涉及一种基于锚点检测的点云补全方法。

技术介绍

[0002]预测完整的3D点云是许多计算机视觉的核心任务。
[0003]现有方法通常基于深度学习，从不完整点云中提取全局特征，用于还原完整点云。尽管它们在一定程度上做到了预测完整点云，但无法准确生成该对象的结构细节，且存在疏密不均的问题。
[0004]将深度学习技术应用于点云数据，是近几年兴起的方向。对于点云补全任务，可以训练神经网络实现目的，即：输入不完整的点云结构，生成完整的点云结构。
[0005]目前，基于深度学习的大多数网络模型对点云补全的效果不佳，其原因是这些模型单纯的将点云补全的任务归类为回归问题，并训练神经网络预测点云坐标，因此只能生成大体上正确，而形状并不规整的点云。为此，有必要实现形状规整，密度均匀的点云补全。

技术实现思路

[0006]为了克服现有技术中局部特征提取的不足，本专利技术提出了一种基于锚点检测的点云补全方法，并设计了与之配合的网络结构，可以生成规整、均匀的点云坐标。实验证明该专利技术在视觉和指标上都有很优秀的效果。
[0007]本专利技术采用的技术方案为：
[0008]一种基于锚点检测的点云补全方法，包括以下步骤：
[0009]S1、数据集获取与数据预处理：获取点云公共数据集ShapeNet，使用局部删除操作，在完整点云数据中删除局部点云，以此构建不完整点云；
[0010]S2、神经网络模型构建：基于由粗略到精细的点云补全思想，先通过锚点的正负性判断得出形状规整密度均匀的稀疏点云，再由稀疏点云的扩散得到密集点云，为实现该思想构建GCNet网络模型；
[0011]S3、模型训练与模型保存：使用Adam优化器训练网络，降低损失函数，提高点云补全效果，当模型的损失函数趋于稳定后，保存模型。
[0012]所述步骤S1中，所述公共数据集ShapeNet，包含16个种类的点云数据，所有种类的每一个单独点云数据文件(.pts)都由三维坐标系中n个点云坐标x、y、z构成，n的值并不相同。
[0013]所述步骤S1中，所述局部删除操作的具体方法有关键点定义与选取、关键点邻近点云删除。
[0014]所述步骤S1中，所述关键点定义与选取的具体方法为：以(1,0,0)、(0,0,1)、(1,0,1)、(
‑
1,0,0)4个点为关键点，()中的三个数字表示三维坐标系中x、y、z的坐标；为模拟缺失不同部位的不完整点云，我们依次使用4个关键点作为中心点，并删除中心点位置附近的点云数据；所述关键点邻近点云删除操作的具体步骤为：计算点云中所有点与中心的距离，
删除最靠近中心点的512个点云。
[0015]所述步骤S2中，所述通过锚点的正负性判断得出形状规整密度均匀的稀疏点云中，由于空间中负样本的数量远远多于正样本，过多的负样本对于网络的训练并无收益，因此为了解决负样本过多的问题，我们在负样本中只选取正样本数量的三倍用于训练。此外，在选中的负样本中，我们通过设置阈值d，进一步将负样本划分为难训练负样本和易训练负样本，距离小于阈值d的为难训练负样本；大于阈值d的易训练负样本，其中难训练负样本与易训练负样本的数量比为2：1。
[0016]所述步骤S2中，所述由稀疏点云的扩散得到密集点云中，由于稀疏点云只能在整体上表示点云的形状结构，而不能显示其具体细节，因此我们在稀疏点云(正锚点样本)的基础上通过神经网络生成各正锚点邻域内的点云坐标，并通过tanh限制邻域的最大范围，使其不超过两个锚点间距的一半。
[0017]所述步骤S2中，所述GCNet包含编码器与解码器两部分，所述编码器部分的具体目标是：提取输入的不完整点云的坐标特征；所述编码器部分的具体结构是：输入点云坐标首先通过一层卷积层，得到点云的特征向量，接着将点云所在空间划分为w
×
w
×
w个锚点，锚点所对应的网格内的向量取平均，以此实现特征的网格结构化，最后通过3D卷积和残差连接提取高维特征；所述解码器部分的具体目标是：通过解码器部分提取的高维特征生成缺失部分的点云坐标；所述解码器部分的具体结构是：输入的高维特征首先通过3D转置卷积，实现特征的上采样，用于预测锚点的正负性和正锚点邻域内的扩散点云坐标信息，生成最终的完整点云。
[0018]所述步骤S3中，所述损失函数包含BCELoss和CD，BCELoss的具体公式为：
[0019][0020]其中X
i
为预测值，y
i
为标签，CD的具体公式为：
[0021][0022]其中CD(S1,S2)表示预测生成的点云集合S1和真实的点云集合S2之间的欧式空间平均最近距离；x、y分别表示点云集合中的一个点云，由x、y、z三个坐标组成。
[0023]本专利技术与现有技术相比所具有的有益效果为：
[0024]本专利技术摒弃了纯回归思路预测点云坐标的思路，训练神经网络GCNet，该神经网络模型能够判断锚点的正负性以及生成锚点扩散的三维坐标信息，实现了端到端的点云补全，并解决了现有方法只依靠神经网络回归生成点云坐标而导致的补全效果不佳的问题。提高了点云补全的效果。
附图说明
[0025]图1为本专利技术的流程图；
[0026]图2为本专利技术GCNet结构图；
[0027]图3为实施例1中飞机点云补全效果；
[0028]图4为实施例2中飞机点云补全效果；
[0029]图5为实施例3中飞机点云补全效果；
具体实施方式
[0030]为使本专利技术实施例的目的、技术方案和优点更加清楚，下面对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0031]一种基于锚点检测的点云补全方法，使用ShapeNet数据集进行点云补全的训练，该数据集包含16类，共有15011个点云数据，其中训练集12137个点云数据，测试集2874个点云数据。对每一个点云数据通过视点操作做预处理得到网络输入的不完整点云。将不完整点云输入GCNet，使用BCELoss和CD损失函数进行训练，100轮训练后保存效果最好的权重。如图1所示，详细过程包括以下步骤：
[0032]S1、数据集获取与数据预处理：获取点云公共数据集ShapeNet，使用局部删除操作，在完整点云数据中删除局部点云，以此构建不完整点云；
[0033]S2、神经网络模型构建：基于由粗略到精细的点云补全思想，先通过锚点的正负性判断得出形状规整密度均匀的稀疏点云，再由稀疏点云的扩散得到密集点云，为实现该思想构建GCNet网络模型；
[0034]S3、模型训练与模型保存：使用Adam优化器训练网络，降低损失函数，提高点云补全效果，当模型的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于锚点检测的点云补全方法，其特征在于：包括以下步骤：S1、数据集获取与数据预处理：获取点云公共数据集ShapeNet，使用局部删除操作，在完整点云数据中删除局部点云，以此构建不完整点云；S2、神经网络模型构建：基于由粗略到精细的点云补全思想，先通过锚点的正负性判断得出形状规整密度均匀的稀疏点云，再由稀疏点云的扩散得到密集点云，为实现该思想构建GCNet网络模型；S3、模型训练与模型保存：使用Adam优化器训练网络，降低损失函数，提高点云补全效果，当模型的损失函数趋于稳定后，保存模型。2.根据权利要求1所述的一种基于锚点检测的点云补全方法，其特征在于：所述步骤S1中，所述公共数据集ShapeNet，包含16个种类的点云数据，所有种类的每一个单独点云数据文件(.pts)都由三维坐标系中n个点云坐标x、y、z构成，n的值并不相同。3.根据权利要求1所述的一种基于锚点检测的点云补全方法，其特征在于：所述步骤S1中，所述局部删除操作的具体方法有关键点定义与选取、关键点邻近点云删除。4.根据权利要求3所述的一种基于锚点检测的点云补全方法，其特征在于：所述步骤S1中，所述关键点定义与选取的具体方法为：以(1,0,0)、(0,0,1)、(1,0,1)、(
‑
1,0,0)4个点为关键点，()中的三个数字表示三维坐标系中x、y、z的坐标；为模拟缺失不同部位的不完整点云，依次使用4个关键点作为中心点，并删除中心点位置附近的点云数据；所述关键点邻近点云删除操作的具体步骤为：计算点云中所有点与中心的距离，删除最靠近中心点的512个点云。5.根据权利要求1所述的一种基于锚点检测的点云补全方法，其特征在于：所述步骤S2中，所述通过锚点的正负性判断得出形状规整密度均匀的稀疏点云中，在负样本中只选取正样本数量的三倍用于训练；此外，在选中的负样...

【专利技术属性】
技术研发人员：邹艳妮，张怡睿，徐嘉伯，刘小平，刘捷，
申请(专利权)人：南昌大学，
类型：发明
国别省市：