一种高准确率的点云数据分类分割改进算法

技术领域

本发明涉及高维点云数据处理技术领域，特别涉及一种高准确率的点云数据分类分割改进算法。

背景技术

点云数据处理在测绘、自动驾驶、农业、规划设计、医疗等方面有着重要作用。在目前的相关方法中，对点云数据的分类分割方法主要有三种：第一种是3D 物体不同“视角”所得到的2D 渲染图，将这些2D图像通过深度学习网络进行特征学习，再通过量化视角的差异，根据视角的差异得分将视角集合划分为若干组，最后，通过组内池化和跨组融合后再用于预测；第二种是将每个点云映射到体素化空间上，然后通过两个三维卷积层学习体素级的特征；第三种是采用神经网络，直接在原始点云数据上进行处理。前两种方法由于需要转换到别的方式，数据处理存在的时间长、准确率不高；第三种减少了信息的损失，但在原始数据上直接对点处理忽略了点与点之间的拓扑信息，导致准确率不高。

综上所述，现有的点云数据的分类分割方法存在的问题点是点云数据处理存在的时间长、准确率不高。

发明内容

针对以上缺陷，本发明的目的是提供一种高准确率的点云数据分类分割改进算法，此高准确率的点云数据分类分割改进算法是基于神经网络，采用融合动态图卷积和DHT差异性池化算法的方法，在提取局部特征时，构建点与点的相互关系，获得更多点与点之间的相关信息，在提取全局特征时，改变池化方法保留更多信息，减少信息损失，提高点云分类分割任务的准确率。

为了实现上述目的，本发明的技术方案是：

一种高准确率的点云数据分类分割改进算法，包括以下步骤：步骤一：输入点云，采集分散的点云数据；步骤二：空间转换网络进行坐标对齐，将步骤一采集到的分散的点云数据通过与空间转换网络训练得到的空间转换矩阵进行坐标对齐，得到可以输入网络的数据格式的点云；步骤三：动态图卷积模块，所述动态图卷积模块中存储有动态图卷积算法，所述动态图卷积算法可以将步骤二获得的所述数据格式的点云通过K-NN的方式获得获得拓扑结构，再对拓扑结构构成图，对该图结构经过图卷积和最大池化，获得每一个采样点与周围点的特征关系；步骤四：DHT差异性池化模块，所述DHT差异性池化模块中存储有DHT差异性池化算法，所述DHT差异性池化算法可以将步骤三获得的特征关系经过离散哈特莱转换到想要的维度，再通过并联的最大池化和均值池化，将并联通道的特征融合作为全局特征；步骤五：全连接层，步骤四得到的特征为高维特征，将步骤四得到的特征通过全连接层降维得到低维特征。步骤六：Softmax多分类，将步骤五输出的特征作为最终特征，将此特征作为得分，通过Softmax回归分类函数进行分类；步骤七：输出得分，完成分类任务。

其中，步骤一中所述的点云数据通过激光雷达设备采集。

其中，步骤三所述的动态图卷积算法包括以下流程：通过k-NN找到采样点的邻域点、获得图结构、MLP多层感知机、获得边信息、最大池化聚合、输出特征。

其中，所述步骤四中DHT差异性池化算法包括以下流程：输入特征、离散哈特莱转换、中心化、剪裁目标大小、哈特莱逆转换，所述哈特莱逆转换包括并联的最大池化通道和均值池化通道，所述最大池化通道和所述均值池化通道输出端融合特征，然后进行输出。

其中，所述步骤五中全连接层包括三个神经元不同连接层。

其中，其特征在于，所述Softmax回归分类函数表达式：

采用了上述技术方案后，本发明的有益效果是：

本发明基于神经网络，采用融合动态图卷积算法和DHT差异性池化算法的方法，极大提升了点云数据分类分割的准确率。在获得局部特征时，传统神经网络将每个点云作为独立点进行卷积处理，忽略了相关点之间的联系，本发明为克服该缺陷，采用动态图卷积算法构建相关点的拓扑结构图进行图卷积处理，提高了对点云数据的表征能力；同时，传统神经网络方法直接使用最大池化获得全局特征，本发明算法先使用哈莱特转换减少了直接池化的损失，减小了直接池化的误差，再通过设计的差异性并联池化通道，弥补了单一最大池化通道的信息局限，保留了不同的特征信息，使池化后的特征更加丰富。

综上所述，本发明高准确率的点云数据分类分割改进算法解决了现有技术中点云数据处理存在的时间长、准确率不高的技术问题，本发明采用差异性并联池化通道，在此基础上，将动态图卷积与哈特莱转换及差异性并联池化融合应用于点云数据分类分割实验中，利用测试结果的数据对比来验证改进后的点云数据分类分割方法较之间方法的准确率更高。

附图说明

图1是本发明一种高准确率的点云数据分类分割改进算法的流程图；

图2是卷积过程示意图；

图3是动态图卷积算法流程图；

图4是DHT差异性池化算法流程图。

具体实施方式

下面结合附图和实施例，进一步阐述本发明。

本说明书中涉及到的方位均以附图所示方位为准，仅代表相对的位置关系，不代表绝对的位置关系。

如图1、图2、图3和图4共同所示，一种高准确率的点云数据分类分割改进算法，其特征在于，包括以下步骤：

步骤一：输入点云，采集分散的点云数据；通过激光雷达等设备采集分散的点云数据。

步骤二：空间转换网络进行坐标对齐。

将步骤一采集到的分散的点云数据通过与Transform Net空间转换网络训练得到的空间转换矩阵进行坐标对齐，得到可以输入网络的数据格式的点云。

Transform Net空间转换网络用于将不同旋转平移的原始点云和点云特征进行规范化，它的主要作用是学习得出变化矩阵来对输入的点云或特征进行规范化处理。

Transform Net空间转换网络经过升维卷积、最大池化、降维卷积实现维度的转换，输出为3*3的空间转换矩阵。卷积运算的数学表达式为（1）所示：

式中：

步骤三：动态图卷积模块，动态图卷积模块中存储有动态图卷积算法。动态图卷积算法可以将步骤二获得的数据格式的点云通过K-NN的方式获得获得拓扑结构，再对拓扑结构构成图，对该图结构经过图卷积和最大池化，获得每一个采样点与周围点的特征关系。

动态图卷积算法包括以下流程：通过k-NN找到采样点的邻域点、获得图结构、MLP多层感知机、获得边信息、最大池化聚合、输出特征。

工作流程如下：输入步骤二获得的数据格式的点云，受图卷积神经网络的启发，通过k-NN找到采样点的邻域点，获得拓扑结构，对所有目标采样点，依次遍历每一个采样点，构成邻域范围，获得图结构，再通过共享权值的多层感知机MLP来获得边信息，有效保留了采样点之间的相互关系；最后将这些特征采用最大池化的方式降维到一维，以此方法实现所有目标点的一次更新，重复该步骤获得不同的图结构完成不同的图卷积为动态图卷积，以动态图卷积获得点云数据的局部特征。

动态图卷积的目标为图结构，图结构是由节点和边构成的不规则结构。本发明中图结构的节点为目标采样点及其邻域点，图结构中的边为目标采样点及其邻域点之间的距离，计算边信息的数学表达式为（2）所示：

式中：

图卷积神经网络算法：

传统的神经网络算法目标是图像、文本、语音等规则的数据，而图卷积神经网络算法的目标是非结构化的数据，这类数据用传统的神经网络算法难以建模，图卷积神经网络算法通过神经网络来聚合每个节点及其周围节点的信息，完成每个节点的更新。每个目标点的更新为聚合其边信息，由于点云数据的无序性，所以选择采用最大池化进行聚合，所有目标点全部更新一遍为完成一层图卷积更新，随着每一层的更新重新构成图结构，再重新使用图卷积，完成动态图卷积。图卷积神经网络算法的公式可以表示如下：

式中：

动态图卷积算法流程：

（1）通过k-NN找到采样点的邻域点，通过k近邻的方式获得获得图结构。第一步：计算目标采样点与点云数据中所有采样点的距离，可以是欧式距离、余弦距离等，比较常用的是较为简单的欧式距离；第二步：找出上步计算的距离中最近的k个对象，作为目标采样点的邻居，公式可以表示如下：

式中：

（2）通过MLP多层感知机，获得边信息。计算边缘信息，计算公式可以表示如下：

式中：

（3）最大池化聚合，输出特征。

单一更新目标点特征操作，公式表示如下：

式中：

所有目标点动态更新，公式表示如下：

式中：

步骤四：DHT差异性池化模块，DHT差异性池化模块中存储有DHT差异性池化算法，DHT差异性池化算法可以将步骤三获得的特征关系经过离散哈特莱转换转换后从空间域转换到频域；再进行中心化将高值固定在频域中心；然后裁剪到设定维度；最后经过逆变换从频域转换到空间域，转换后的特征通过并联的最大池化通道和均值池化通道，获得维度相同的特征，但这两个特征保留不同的信息；最后将并联通道的特征融合作为全局特征。

此方法获得的全局特征损失的信息较少，包含的信息更加丰富。

DHT差异性池化算法包括以下流程：输入特征、离散哈特莱转换、中心化、剪裁目标大小、哈特莱逆转换，所述哈特莱逆转换包括并联的最大池化通道和均值池化通道，所述最大池化通道和所述均值池化通道输出端融合特征，然后进行输出。

并联池化通道一条选择最大池化，最大池化是将目标维度的数据取最大值作为该维度的特征值，并联池化通道另一条选择均值池化，均值池化是将目标维度的数据取均值作为该维度的特征值。

将特征采用离散哈莱特转换从空间域转换到频域，转换到频域的远点的周围区域包含了最高值，但由于其周期性，频域的4个角包也会出现类似的高值，所以要再将其中心化，接着按照想要的维度裁剪中心大小，最后再利用离散哈莱特逆变换转换回空间域。

（1）离散哈特莱转换，公式表示如下：

式中：

（2）中心化，公式表示如下：

式中：

（3）离散哈特莱逆转换，公式表示如下：

差异性池化算法：

（1）最大池化

由于点云数据的无序性，所以并联池化通道要选择对称函数。对于输入点云X ,最大池化的全局特称向量为：

式中：

（2）均值池化

对于输入点云X，均值池化的全局特称向量为：

式中：

（3）差异性池化

最大池化和均值池化可以获得不同的特征信息，再将不同的特征向量并联得到全局特征，表示为：

式中：

步骤五：全连接层，步骤四得到的特征为高维特征，将步骤三得到的特征通过通过三个神经元不同的全连接层，逐步降维得到低维特征。

全连接层公式表示如下：

式中：f表示激活函数；n为

步骤六：Softmax多分类，将步骤五输出的特征作为该点云数据经过网络处理后最终的特征，输出的特征作为将此特征作为得分，通过Softmax回归分类函数进行分类，将预测特征与真实特征比较得到分类结果，完成分类任务。

Softmax回归分类

Softmax回归是一个线性的多分类模型，是Logistic回归模型的延伸。在多分类问题上，可以把真实权重向量转化为一种概率分布，Softmax回归分类函数表达式：

式中：M为样本个数；N为种类个数；为第i个样本的特征向量；为种类标识；和分别作为全连接层的权值矩阵和偏移量；为权值矩阵的第j列；为对应的偏置项。

步骤七：输出得分，将预测特征与真实特征比较得到分类结果，完成分类任务。

本发明所述的是一种高准确率的点云数据分类分割改进算法，为点云分类、点云部件分割和点云语义分割提供一种新的、高准确率的算法，主要通过图卷积模块构建点与点之间的相互关系，同时采用动态更新图结构的方式，增强网络表征能力。采用哈特莱转换减少了直接池化的损失，同时，采用差异性并联池化通道，获得更多信息，提高网络分类分割精度，同时使网络具有更强的鲁棒性。实验结果和性能表明，基于动态图卷积和DHT差异性池化的点云数据分类分割算法在点云分类任务、点云部件分割任务和点云语义分割任务有更高的准确，在点云数据处理领域具有重要的意义。

本发明不局限于上述具体的实施方式，本领域的普通技术人员从上述构思出发，不经过创造性的劳动，所做出的种种变换，均落在本发明的保护范围之内。