[cvpr17]Multi-View 3D Object Detection Network for Autonomous Driving
3D点云做detection的一篇milestone paper。经典的two-stage方法(region proposal-based method)。思路来自于经典的faster rcnn。
整个模型如下图
图一. 整体模型
3D Point Cloud Representation
这篇文章可以归纳为是multiview-based method,multi-view的方法是指将三维点云按照不同view进行映射,得到很多的2D图像,因为2D图像的CNN很强大,这样就可以用2D CNN处理。最后将不同view得到的feature进行融合(fusion),再进行后续的不同task,如classification、detection、segmentation等等。这种方法优缺点都很明显,优点是CNN处理2D图像很强大也更成熟,缺点是有限的views会造成信息的损失。
这篇文章映射的角度包括Bird's Eye View(鸟瞰图)和Front View(前视图)。
Bird’s Eye View Representation.
对整个点云按照鸟瞰的视角,按照0.1米的分辨率进行网格化(grid)采样。每个网格(cell)在高度这一特征(height feature)上取该网格里所有点云的最大值。特别地,作者并不是仅仅只取一个鸟瞰图,而是对整个点云按照不同高度,均匀切片成M片(M slices)。每个切片产生一张height map,里面记录了每个切片里面每个cell的最大高度。采样的每个网格(cell)的intensity feature就是max height的那个点的intensity。点云的density feature指每个cell的点数量(记为$N$),按照$min(1.0, \frac{log(N+1}{log(64)})$进行归一化。因此共有M个height map,1个density feature和1个intensity feature,共$N+2$维的特征。
Front View Representation.
前视图作为鸟瞰图的补充信息。但是由于点云数据稀疏程度是不均匀的,很多地方非常稀疏,因此作者并不是将点云投射到图像平面(image plane),而是投射到圆柱平面(cylinder plane)来产生密集的前视图。
极坐标投影转换关系如下图,得到前视图极坐标$P_{f_c}=(r,c)$
图二. 前视图圆柱投影
3D Proposal Network
首先用一个网络生成region proposals. 该文用鸟瞰图作为输入,由3D prior boxes生成3D proposal boxes。3D box定义为$(x,y,z,l,w,h)$,分别代表中心点三维坐标和尺寸。鸟瞰图中的anchor box可由3D box离散化得到,即$x_{bv}, y_{bv}, l_{bv}, w_{bv}$。在faster RCNN中,每个像素点产生的anchor有三种大小、三种比例共计九个,而在该文里prior box即anchor,针对车辆检测问题,prior box的$(l,w)$设定为{(3.9,1.6),(1.0,1.6)}两种(聚类得到),高h设定为1.56m。对prior box(按Z轴)旋转90°,共得到N=4种prior boxes (两种大小,两种角度即横竖摆放)。鸟瞰图中$(x,y)$是变化的,$z$则由相机高度和物体高度计算得到。
由于0.1米的采样距离使得车辆只占5~40个像素,小目标很难被发现,因此网络中有一个上采样的过程,以保证feature map足够大。
3D box的回归是回归$t=(\delta{x}, \delta{y}, \delta{z}, \delta{l},\delta{w},\delta{h}$。同样用cross-entropy loss做分类,smooth $l_1$ loss 做回归。anchor和ground truth的IoU>0.7就是positive,IoU<0.5为negative。
Region-based Fusion Network
结合不同view,做object proposals的分类和坐标回归。
Multi-View ROI Pooling.
对于每个产生的3D proposal, 分别求鸟瞰图、前视图(圆柱投影)和Image plane RGB图,然后分别求ROI polling得到特征。
Deep Fusion.
特征融合的方式,如下图
图三. deep fusion for multi-view.
Oriented 3D Box Regression
在stage-two,回归的坐标发生了变化,直接回归3D box的八个顶点的坐标。作者表明尽管24D向量存在冗余,但实际效果更好。这样,box的方位就直接由八个点确定,不需要再回归。
Network Regularization