PointRCNN论文解读和代码解析

CVplayer111

已于 2022-10-22 15:21:50 修改

阅读量1.4k

点赞数 3

分类专栏：三维目标检测论文阅读文章标签：深度学习目标检测计算机视觉

于 2022-10-20 09:09:42 首次发布

本文链接：https://blog.csdn.net/slamer111/article/details/127407703

版权

三维目标检测论文阅读专栏收录该内容

11 篇文章 4 订阅

订阅专栏

论文地址：https://arxiv.org/abs/1812.04244

代码地址：GitHub - open-mmlab/OpenPCDet: OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

一、论文动机

3D目标检测现在比较SOTA的方法有两种，一种是将点云投影到俯视图上，使用2D的检测框架。要么是变成规则的Voxel，这两种方法在量化过程中都存在信息缺失的问题。而且最近有人提出了pointnet和pointnet++这种直接从点云中学习特征的网络，所以作者想借鉴这两个，直接从3D点云中获取检测框。同时3D场景中物体都是自然分割的，相互之间没有重叠，所以3D物体的分割掩码都能通过他们的3D真实框直接获取，框内的点被认为前景点。

二、论文方法

提出了一个两阶段的Anchor-free3D目标检测框架，PointRCNN首次直接在点云数据上分割mask

第一阶段用一个自底向上的方法生成3D候选框，先分割前景点，同时在这些被分割出来的点中生成少些候选框。它的优势是避免了在整个3D空间使用大量的锚框。

第二个阶段是对生成的3D候选框进行微调。将局部框内xyz和depth深度信息，分割掩码进行MLP升维，然后和第一阶段的点特征结合再MLP分类回归。

它的优势是充分利用了第一阶段分割和推荐子网络提供的全部信息。

为了学习更有效的坐标微调，作者还提出了针对推荐生成和微调的完全基于bin的3D box回归损失，加速了网络的收敛速度和准确率。

三、网络结构

第一阶段

我们通过逐点地学习特征来分割原始点云，并且从分割好的前景点中生成3D候选框，这种方法我们避免了设置大量的3D锚框，极大的限制了3D候选框的生成候选空间。为了学习有区别的逐点特征来描述原始点云，我们使用带有多尺度分组的PointNet++作为骨干网络，后面增加了一个估计前景掩码的分割头和一个生成3D候选框的回归头，同时进行前景分割和候选框回归。对于大尺度的外景，前景点数量远小于背景点，所以我们使用Focal Loss.

虽然只是从前景点回归3D回归框位置，没有从背景点回归，但因为点云网络的感受野，背景点也为box的生成提供了信息。为了估计一个目标的中心位置，我们将每个前景点周围的xz轴分割成一系列离散的bin，我们发现对x-z轴使用基于bin的交叉熵损失分类比直接smoothL1 LOSS回归能得出更加精确鲁棒的中心定位。而y轴直接L1 Loss就可以获得精确的y值。物体的方向估计也是基于bin的，hwl则是直接由整个训练集每个类的平均尺寸直接回归得到

为了移除冗余的候选框，我们对鸟瞰图的定向IOU使用NMS，来生成少些高质量的候选框，训练时我们将IOU设置为0.85，NMS保留前300个给二阶段。推理时，设置IOU阈值为0.8，NMS保留前100个给二阶段。

第二阶段

为了学习每个候选框更加具体的局部特征，我们根据每个推荐框来池化(就是降采样)它的3D点以及对应特征。我们对候选框进行扩大，来从旁边获取有用的环境信息，对于每一个点判断在不在扩大的框里，在的话这个点和他的特征就被保留用来微调候选框。后面框内坐标统一转换为正则坐标系下的局部坐标，因为这样可以更好的学习每个proposals的局部特征。对于每个候选框内的点，他的局部空间特征，分割掩码和空间距离经过MLP与第一阶段点特征结合，喂进PointNet++网络进行置信度和box细化。