Command Palette
Search for a command to run...
基于中心的3D目标检测与跟踪
基于中心的3D目标检测与跟踪
Tianwei Yang Jiajun Han Yonghao Li Zhenliang Ma Xiaopeng Ji Xinggang Wang
摘要
在点云中,三维物体通常被表示为3D框。这种表示方法模仿了广泛研究的基于图像的2D边界框检测,但带来了额外的挑战。三维世界中的物体没有特定的方向,而基于框的检测器难以枚举所有方向或将轴对齐的边界框拟合到旋转的物体上。本文提出了一种将3D物体表示、检测和跟踪为点的方法。我们的框架——CenterPoint,首先使用关键点检测器检测物体中心,并回归其他属性,包括3D尺寸、3D方向和速度。在第二阶段,它利用物体上的额外点特征来细化这些估计值。在CenterPoint中,3D物体跟踪简化为贪婪最近点匹配。由此产生的检测和跟踪算法简单、高效且有效。CenterPoint在nuScenes基准测试中实现了3D检测和跟踪的最佳性能,单模型分别达到了65.5 NDS(NuScenes Detection Score)和63.8 AMOTA(Average Multi-category Tracking Accuracy)。在Waymo开放数据集上,CenterPoint大幅超越了所有先前的单模型方法,并在所有仅使用激光雷达的提交中排名第一。代码和预训练模型可在https://github.com/tianweiy/CenterPoint 获取。