arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.04634cs.CVcs.LGcs.ROstat.ML

基于能量模型的高精度3D目标检测

Accurate 3D Object Detection using Energy-Based Models

Uppsala University(乌普萨拉大学);ETH Zürich(苏黎世联邦理工学院)

Fredrik K. Gustafsson, Martin Danelljan, Thomas B. Schön

AI总结:

针对稀疏LiDAR下3D目标检测精度不足的问题,提出适配3D边界框的可微分池化算子构建EBM网络,集成至SA-SSD后在KITTI数据集上全指标优于基线,验证了EBM回归在高精度3DOD中的潜力。

摘要:

高精度3D目标检测(3DOD)对自主机器人在复杂环境中安全导航至关重要。然而,在杂乱环境中基于稀疏LiDAR数据回归精准3D边界框是一项极具挑战性的任务。我们通过探索条件能量基模型(EBMs,Energy-Based Models)在概率回归领域的最新进展来解决该任务。尽管采用EBM进行回归的方法已在图像2D目标检测上展现出优异性能,但这些技术无法直接应用于3D边界框。因此,本研究设计了一种适用于3D边界框的可微分池化算子,作为我们EBM网络的核心模块。我们进一步将该通用方法集成到最先进的3D目标检测器SA-SSD中。在KITTI数据集上,我们提出的方法在所有3DOD指标上均持续优于SA-SSD基线,证明了基于EBM的回归在实现高精度3DOD方面的潜力。代码可在https://github.com/fregu856/ebms_3dod获取。

英文摘要:

Accurate 3D object detection (3DOD) is crucial for safe navigation of complex environments by autonomous robots. Regressing accurate 3D bounding boxes in cluttered environments based on sparse LiDAR data is however a highly challenging problem. We address this task by exploring recent advances in conditional energy-based models (EBMs) for probabilistic regression. While methods employing EBMs for regression have demonstrated impressive performance on 2D object detection in images, these techniques are not directly applicable to 3D bounding boxes. In this work, we therefore design a differentiable pooling operator for 3D bounding boxes, serving as the core module of our EBM network. We further integrate this general approach into the state-of-the-art 3D object detector SA-SSD. On the KITTI dataset, our proposed approach consistently outperforms the SA-SSD baseline across all 3DOD metrics, demonstrating the potential of EBM-based regression for highly accurate 3DOD. Code is available at https://github.com/fregu856/ebms_3dod.

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

↑