arXiv:2411.12177cs.CV2024-11被引 9

无需传感器校准,用注意力机制实现高效3D语义占据预测

Robust 3D Semantic Occupancy Prediction with Calibration-free Spatial Transformation

  • 用纯注意力机制替代校准依赖的2D到3D投影
  • 在三个数据集上超越现有方法,速度提升19.8倍,几何交并比提高1.1
  • 适合需要轻量实时推理的自动驾驶系统

3D语义占据预测对自动驾驶系统至关重要,需将多传感器信息融合至统一3D空间。现有方法依赖传感器校准进行2D到3D变换,但易受校准误差影响且计算开销大。本文提出无需校准的鲁棒高效占据预测框架REO:基于原始注意力机制隐式建模空间对应关系,直接将2D特征投影至预定义俯视图平面;引入2D与3D辅助任务增强骨干网络对空间、语义和纹理特征的区分能力;采用查询式预测策略高效生成大尺度细粒度占据结果。通过融合点云提供互补空间信息,REO在OpenOccupancy、Occ3D-nuScenes和SemanticKITTI Scene Completion三个基准上显著领先。例如,在OpenOccupancy上相比Co-Occ实现19.8倍加速,几何交并比提升1.1。

原文摘要 · Abstract (English)

3D semantic occupancy prediction, which seeks to provide accurate and comprehensive representations of environment scenes, is important to autonomous driving systems. For autonomous cars equipped with multi-camera and LiDAR, it is critical to aggregate multi-sensor information into a unified 3D space for accurate and robust predictions. Recent methods are mainly built on the 2D-to-3D transformation that relies on sensor calibration to project the 2D image information into the 3D space. These methods, however, suffer from two major limitations: First, they rely on accurate sensor calibration and are sensitive to the calibration noise, which limits their application in real complex environments. Second, the spatial transformation layers are computationally expensive and limit their running on an autonomous vehicle. In this work, we attempt to exploit a Robust and Efficient 3D semantic Occupancy (REO) prediction scheme. To this end, we propose a calibration-free spatial transformation based on vanilla attention to implicitly model the spatial correspondence. In this way, we robustly project the 2D features to a predefined BEV plane without using sensor calibration as input. Then, we introduce 2D and 3D auxiliary training tasks to enhance the discrimination power of 2D backbones on spatial, semantic, and texture features. Last, we propose a query-based prediction scheme to efficiently generate large-scale fine-grained occupancy predictions. By fusing point clouds that provide complementary spatial information, our REO surpasses the existing methods by a large margin on three benchmarks, including OpenOccupancy, Occ3D-nuScenes, and SemanticKITTI Scene Completion. For instance, our REO achieves 19.8$\times$ speedup compared to Co-Occ, with 1.1 improvements in geometry IoU on OpenOccupancy. Our code will be available at https://github.com/ICEORY/REO.

3D占据自动驾驶注意力机制多传感器融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。