arXiv:2410.24001cs.CV2024-10NeurIPS被引 21

仅用2D图像训练,实现开放词汇3D目标检测

ImOV3D: Learning Open-Vocabulary Point Clouds 3D Object Detection from Only 2D Images

  • 用单目深度估计和渲染构建伪多模态表示,弥合2D与3D差距
  • 在SUNRGBD和ScanNet上超越现有方法,无需真实3D数据即有效
  • 适合缺乏3D标注数据但有丰富2D图像的场景

开放词汇3D目标检测(OV-3Det)旨在超越训练阶段标注的有限类别。其最大瓶颈是3D标注数据稀缺,而2D图像数据集丰富且标注详尽。因此,利用2D图像中的丰富标注可缓解OV-3Det的数据不足问题。本文探索仅使用2D图像学习OV-3Det的极限。主要挑战在于训练图像与测试点云之间的模态鸿沟,阻碍2D知识向3D检测的有效迁移。为此,我们提出ImOV3D框架,通过包含图像与点云(PC)的伪多模态表示来缩小模态差距。核心在于灵活的模态转换:2D图像可通过单目深度估计升维为3D,也可通过渲染从3D场景生成。这使训练图像与测试点云统一到共享的图像-点云表示中,融合大量2D语义信息,并保留3D空间数据的深度与结构特性。我们精心设计转换过程,最小化训练与测试间的域差异。在两个基准数据集SUNRGBD和ScanNet上的大量实验表明,即使无真实3D训练数据,ImOV3D也显著优于现有方法;加入少量真实3D数据微调后,性能更大幅超越此前最先进水平。代码与预训练模型已发布于https://github.com/yangtiming/ImOV3D。

原文摘要 · Abstract (English)

Open-vocabulary 3D object detection (OV-3Det) aims to generalize beyond the limited number of base categories labeled during the training phase. The biggest bottleneck is the scarcity of annotated 3D data, whereas 2D image datasets are abundant and richly annotated. Consequently, it is intuitive to leverage the wealth of annotations in 2D images to alleviate the inherent data scarcity in OV-3Det. In this paper, we push the task setup to its limits by exploring the potential of using solely 2D images to learn OV-3Det. The major challenges for this setup is the modality gap between training images and testing point clouds, which prevents effective integration of 2D knowledge into OV-3Det. To address this challenge, we propose a novel framework ImOV3D to leverage pseudo multimodal representation containing both images and point clouds (PC) to close the modality gap. The key of ImOV3D lies in flexible modality conversion where 2D images can be lifted into 3D using monocular depth estimation and can also be derived from 3D scenes through rendering. This allows unifying both training images and testing point clouds into a common image-PC representation, encompassing a wealth of 2D semantic information and also incorporating the depth and structural characteristics of 3D spatial data. We carefully conduct such conversion to minimize the domain gap between training and test cases. Extensive experiments on two benchmark datasets, SUNRGBD and ScanNet, show that ImOV3D significantly outperforms existing methods, even in the absence of ground truth 3D training data. With the inclusion of a minimal amount of real 3D data for fine-tuning, the performance also significantly surpasses previous state-of-the-art. Codes and pre-trained models are released on the https://github.com/yangtiming/ImOV3D.

3D检测开放词汇2D图像点云

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