OVM3D-Det

Training an Open-Vocabulary Monocular 3D Object Detection Model without 3D Data

NeurIPS 2024
1Tsinghua University, 2NVIDIA Research, 3Stanford University, 4Beijing Academy of Artificial Intelligence,
*Corresponding author
contact: hr20 (at) mails (dot) tsinghua (dot) edu (dot) cn

Abstract

Open-vocabulary 3D object detection has recently attracted considerable attention due to its broad applications in autonomous driving and robotics, aiming to recognize novel classes in previously unseen domains. However, existing point cloud-based models are limited by their high deployment costs. In this work, we propose a novel open-vocabulary monocular 3D object detection framework, dubbed OVM3D-Det, which trains detectors using only RGB images, making it both cost-effective and scalable to publicly available data. Unlike traditional methods, OVM3D-Det does not require high-precision LiDAR or 3D sensor data for either input or generating 3D bounding boxes. Instead, it employs open-vocabulary 2D models and pseudo-LiDAR to automatically label 3D objects in RGB images, fostering the learning of open-vocabulary monocular 3D detectors.

teaser-fig.

Challenge

However, training 3D models with labels directly derived from pseudo-LiDAR is inadequate due to imprecise boxes estimated from noisy point clouds and severely occluded objects. To address these issues, we introduce two innovative designs: adaptive pseudo-LiDAR erosion and bounding box refinement with prior knowledge from large language models. These techniques effectively calibrate the 3D labels and enable RGB-only training for 3D detectors. Extensive experiments demonstrate the superiority of OVM3D-Det over baselines in both indoor and outdoor scenarios.

teaser-fig.

Methodology

teaser-fig.

The overall framework of OVM3D-Det. Step 1: Generate per-instance pseudo-LiDAR. Step 2: Apply an adaptive erosion process to remove artifacts and noises. Step 3: Estimate the orientation. Step 4: Tightly fit a box and utilize object priors to assess the estimated box; if deemed unreasonable, search for the optimal box. Step 5: Train the model with pseudo labels.

Results

Open-vocabulary monocular 3D object detection results on indoor datasets

teaser-fig.

Open-vocabulary monocular 3D object detection results on outdoor datasets

teaser-fig.

Visualizations

teaser-fig.
teaser-fig.

BibTeX


      @inproceedings{huang2024training,
          title={Training an Open-Vocabulary Monocular 3D Detection Model without 3D Data},
          author={Rui Huang and Henry Zheng and Yan Wang and Zhuofan Xia and Marco Pavone and Gao Huang},
          booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
          year={2024},
          url={https://openreview.net/forum?id=EFkw0OgZOr}
      }