用深度预测构建3D几何表示,提升图像的3D目标检测性能
3DGeoDet: General-purpose Geometry-aware Image-based 3D Object Detection
- 基于预测深度生成显式体素和隐式距离函数双重3D表征
- 在SUN RGB-D等多数据集上提升9.3~0.19点检测精度
- 无需3D标注即可实现端到端训练,适合跨环境应用
本文提出3DGeoDet,一种通用的基于图像的3D目标检测方法,适用于室内外单视图与多视图场景。图像类3D检测面临缺乏3D几何线索的问题,导致图像与3D结构对应模糊。为此,3DGeoDet利用预测深度信息,以显式和隐式方式构建高效3D几何表征:通过体素占据注意力显式优化体素化特征体积,并融合截断符号距离函数(TSDF)作为隐式表示。无需依赖3D信号监督,仅通过中间3D表征即显著增强模型对3D几何的理解,实现端到端训练。在多个基准数据集上均超越现有方法,在SUN RGB-D上提升9.3 [email protected],ScanNetV2上提升3.3 [email protected],KITTI上提升0.19 [email protected]。
原文摘要 · Abstract (English)
This paper proposes 3DGeoDet, a novel geometry-aware 3D object detection approach that effectively handles single- and multi-view RGB images in indoor and outdoor environments, showcasing its general-purpose applicability. The key challenge for image-based 3D object detection tasks is the lack of 3D geometric cues, which leads to ambiguity in establishing correspondences between images and 3D representations. To tackle this problem, 3DGeoDet generates efficient 3D geometric representations in both explicit and implicit manners based on predicted depth information. Specifically, we utilize the predicted depth to learn voxel occupancy and optimize the voxelized 3D feature volume explicitly through the proposed voxel occupancy attention. To further enhance 3D awareness, the feature volume is integrated with an implicit 3D representation, the truncated signed distance function (TSDF). Without requiring supervision from 3D signals, we significantly improve the model's comprehension of 3D geometry by leveraging intermediate 3D representations and achieve end-to-end training. Our approach surpasses the performance of state-of-the-art image-based methods on both single- and multi-view benchmark datasets across diverse environments, achieving a 9.3 [email protected] improvement on the SUN RGB-D dataset, a 3.3 [email protected] improvement on the ScanNetV2 dataset, and a 0.19 [email protected] improvement on the KITTI dataset. The project page is available at: https://cindy0725.github.io/3DGeoDet/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。