PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

PointNet++

核心问题:点云的分类/语义分割

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

1、网络构成

 

PointNet提取特征的方式是对所有点云数据提取了一个全局的特征(见上图),显然,这和目前流行的CNN逐层提取局部特征的方式不一样。受到CNN的启发,作者提出了PointNet++,它能够在不同尺度提取局部特征,通过多层网络结构得到深层特征。PointNet++由以下几个关键部分构成:

1.1采样层(sampling)

激光雷达单帧的数据点可以多达100k个,如果对每一个点都提取局部特征,计算量是非常巨大的。因此,作者提出了先对数据点进行采样。作者使用的采样算法是最远点采样(farthest point sampling, FPS),相对于随机采样,这种采样算法能够更好地覆盖整个采样空间。

1.2组合层(grouping)

为了提取一个点的局部特征,首先需要定义这个点的“局部”是什么。一个图片像素点的局部是其周围一定曼哈顿距离下的像素点,通常由卷积层的卷积核大小确定。同理,点云数据中的一个点的局部由其周围给定半径划出的球形空间内的其他点构成。组合层的作用就是找出通过采样层后的每一个点的所有构成其局部的点,以方便后续对每个局部提取特征。

1.3特征提取层(feature learning)

因为PointNet给出了一个基于点云数据的特征提取网络,因此可以用PointNet对组合层给出的各个局部进行特征提取来得到局部特征。值得注意的是,虽然组合层给出的各个局部可能由不同数量的点构成,但是通过PointNet后都能得到维度一致的特征(由上述K值决定)。

1.4网络结构总结:

上述各层构成了PointNet++的基础处理模块。如果将多个这样的处理模块级联组合起来,PointNet++就能像CNN一样从浅层特征得到深层语义特征。对于分割任务的网络,还需要将下采样后的特征进行上采样,使得原始点云中的每个点都有对应的特征。这个上采样的过程通过最近的k个临近点进行插值计算得到。完整的PointNet++的网络示意图如下图所示。

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

 

2、不均匀点云数据的特征提取

不同于图片数据分布在规则的像素网格,且有均匀的数据密度,点云数据在空间中的分布是不规则且不均匀的。虽然PointNet能够用于对各个点云局部提取特征,但是由于点云在各个局部均匀性不一致,很可能导致学习到的PointNet不能提取到很好的局部特征。比如说,在越远的地方激光雷达数据通常变得越稀疏,因此在稀疏的地方应该考虑更大的尺度范围来提取特征。为此,作者提出了两种组合策略来保证更优的特征提取

2.1多尺度组合(multi-scale grouping, MSG):

比较直接的想法是对不同尺度的局部提取特征并将它们串联在一起,如下图(a)所示。但是因为需要对每个局部的每个尺度提取特征,其计算量的增加也是很显著的。

2.2多分辨率组合(multi-resolution grouping, MRG)

为了解决MSG计算量太大的问题,作者提出了MRG。此种方法在某一层对每个局部提取到的特征由两个向量串联构成,如下图(b)所示。第一部分由其前一层提取到的特征再次通过特征提取网络得到,第二部分则通过直接对这个局部对应的原始点云数据中的所有点进行特征提取得到。

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

 

3、其它要点

3.1 FPS最远点采样

 

3.2上采样

在网络中输入不断被降采样而在segmentation中label都是针对原始点的,所以需要对点云进行上采样。作者使用基于k最近邻的反距离加权平均法方法再和之前的set abstraction 中的feature 做一个 concatenate。

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

 

4、实验效果

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space论文解读

 

5、总结

PointNet++对输入点集的特征提取有一定效果,并能学习与距离度量有关的层次化特征。为了处理非均匀点采样问题,考虑了空间中点与点之间的距离度量,作者对此提出了两种解决策略:MSGMRG。通过层级结构利用局部区域信息学习特征,网络结构更有效更鲁棒。