arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2205.07403cs.CV
收藏

PillarNet:基于柱的实时的高性能三维目标检测

PillarNet: Real-Time and High-Performance Pillar-based 3D Object Detection

Harbin Institute of Technology(哈尔滨工业大学);Shanghai Jiao Tong University(上海交通大学)

Guangsheng Shi, Ruifeng Li, Chao Ma

摘要:

实时的【形】和高性能的3D目标检测对于自动驾驶至关重要。最近表现最好的3D物体检测器主要依赖于基于点或基于3D体素的卷积,这两者对于机载部署来说都是计算效率低下的。相比之下,基于柱的方法仅使用2D卷积,这消耗较少的计算资源,但它们在检测精度方面远远落后于基于体素的方法。本文通过研究柱体检测器和体素检测器之间的主要性能差距,提出了一种实时的【形】高性能柱体检测器PillarNet,该PillarNet由一个用于有效柱体特征学习的编码器网络、一个用于空间语义特征融合的颈网络和常用的检测头组成。PillarNet仅使用2D卷积,可灵活选择柱大小,并与经典的2D CNN主干兼容,如VGGNet和ResNet。此外,PillarNet受益于我们设计的方向解耦IoU回归损失以及IoU感知预测分支。在大规模nuScenes数据集和Waymo开放数据集上的大量实验结果表明,所提出的PillarNet在有效性和效率方面优于最先进的3D检测器。代码可在\url{https://github.com/agent-sgs/PillarNet}上获得。

英文摘要:

Real-time and high-performance 3D object detection is of critical importance for autonomous driving. Recent top-performing 3D object detectors mainly rely on point-based or 3D voxel-based convolutions, which are both computationally inefficient for onboard deployment. In contrast, pillar-based methods use solely 2D convolutions, which consume less computation resources, but they lag far behind their voxel-based counterparts in detection accuracy. In this paper, by examining the primary performance gap between pillar- and voxel-based detectors, we develop a real-time and high-performance pillar-based detector, dubbed PillarNet.The proposed PillarNet consists of a powerful encoder network for effective pillar feature learning, a neck network for spatial-semantic feature fusion and the commonly used detect head. Using only 2D convolutions, PillarNet is flexible to an optional pillar size and compatible with classical 2D CNN backbones, such as VGGNet and ResNet. Additionally, PillarNet benefits from our designed orientation-decoupled IoU regression loss along with the IoU-aware prediction branch. Extensive experimental results on the large-scale nuScenes Dataset and Waymo Open Dataset demonstrate that the proposed PillarNet performs well over state-of-the-art 3D detectors in terms of effectiveness and efficiency. Code is available at \url{https://github.com/agent-sgs/PillarNet}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑