arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.09613cs.CV

ImageNet3D:迈向通用对象级3D理解

ImageNet3D: Towards General-Purpose Object-Level 3D Understanding

Johns Hopkins University(约翰斯·霍普金斯大学);Tsinghua University(清华大学);Tongji University(同济大学);University of Freiburg(弗赖堡大学);Max Planck Institute for Informatics(马克斯·普朗克信息学研究所)

Wufei Ma, Guanning Zeng, Guofeng Zhang, Qihao Liu, Letian Zhang, Adam Kortylewski, Yaoyao Liu, Alan Yuille

AI总结:

本文提出ImageNet3D数据集,为ImageNet的200类补充2D框、3D姿态、位置及含3D信息的标注,并通过多项3D任务评估,推动通用对象级3D理解。

摘要:

一个具备通用对象级3D理解能力的视觉模型,应当能够推断自然图像中任意刚性物体的2D信息(例如类别名称和边界框)与3D信息(例如3D位置和3D视点)。这是一项具有挑战性的任务,因为它涉及从2D信号中推断3D信息,而且最重要的是,需要泛化到来自未见过类别的刚性物体。然而,现有的带有对象级3D标注的数据集往往受限于类别数量或标注质量。在这些数据集上开发的模型会成为针对某些类别或领域的专家模型,并且无法实现泛化。在本工作中,我们提出了ImageNet3D,这是一个用于通用对象级3D理解的大型数据集。ImageNet3D为ImageNet数据集中的200个类别补充了2D边界框、3D姿态、3D位置标注,以及穿插有3D信息的图像说明。借助ImageNet3D中提供的新标注,我们可以:(i)分析视觉基础模型的对象级3D感知能力;(ii)研究并开发通用模型,以推断自然图像中任意刚性物体的2D和3D信息;(iii)将统一3D模型与大型语言模型集成,用于3D相关推理。除标准分类和姿态估计外,我们还考虑了两个新任务,即对象级3D感知探测和开放词汇姿态估计。在ImageNet3D上的实验结果证明了我们的数据集在构建具有更强通用对象级3D理解能力的视觉模型方面的潜力。

英文摘要:

A vision model with general-purpose object-level 3D understanding should be capable of inferring both 2D (e.g., class name and bounding box) and 3D information (e.g., 3D location and 3D viewpoint) for arbitrary rigid objects in natural images. This is a challenging task, as it involves inferring 3D information from 2D signals and most importantly, generalizing to rigid objects from unseen categories. However, existing datasets with object-level 3D annotations are often limited by the number of categories or the quality of annotations. Models developed on these datasets become specialists for certain categories or domains, and fail to generalize. In this work, we present ImageNet3D, a large dataset for general-purpose object-level 3D understanding. ImageNet3D augments 200 categories from the ImageNet dataset with 2D bounding box, 3D pose, 3D location annotations, and image captions interleaved with 3D information. With the new annotations available in ImageNet3D, we could (i) analyze the object-level 3D awareness of visual foundation models, and (ii) study and develop general-purpose models that infer both 2D and 3D information for arbitrary rigid objects in natural images, and (iii) integrate unified 3D models with large language models for 3D-related reasoning.. We consider two new tasks, probing of object-level 3D awareness and open vocabulary pose estimation, besides standard classification and pose estimation. Experimental results on ImageNet3D demonstrate the potential of our dataset in building vision models with stronger general-purpose object-level 3D understanding.

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

↑