用视觉语言模型提升室内场景的空间理解能力。
ROOT: VLM based System for Indoor Scene Understanding and Beyond
- 基于GPT-4V迭代检测物体,结合基础模型获取边界框等元信息。
- 构建SceneVLM生成带距离信息的层级场景图,支持空间关系推理。
- 适用于3D重建与具身智能,数据集超61万张图像,代码开源。
视觉语言模型(VLMs)虽取得显著进展,但在室内场景的空间层次推理方面仍面临挑战。本文提出ROOT系统,首先利用GPT-4V开发迭代物体感知算法,检测室内场景中的物体实体;随后通过视觉基础模型获取额外元信息(如边界框)。在此基础上,提出专用的SceneVLM模型,可生成包含距离信息的层级场景图,增强对室内空间布局的理解。为训练SceneVLM,从多个公开室内数据集收集超过61万张图像,并设计半自动化数据生成管道,建立物体间关系并估算距离。通过多种训练策略完成模型训练。实验表明,ROOT在室内场景理解任务中表现优异,且在3D场景生成、具身智能等下游应用中有效。代码将发布于https://github.com/harrytea/ROOT。
原文摘要 · Abstract (English)
Recently, Vision Language Models (VLMs) have experienced significant advancements, yet these models still face challenges in spatial hierarchical reasoning within indoor scenes. In this study, we introduce ROOT, a VLM-based system designed to enhance the analysis of indoor scenes. Specifically, we first develop an iterative object perception algorithm using GPT-4V to detect object entities within indoor scenes. This is followed by employing vision foundation models to acquire additional meta-information about the scene, such as bounding boxes. Building on this foundational data, we propose a specialized VLM, SceneVLM, which is capable of generating spatial hierarchical scene graphs and providing distance information for objects within indoor environments. This information enhances our understanding of the spatial arrangement of indoor scenes. To train our SceneVLM, we collect over 610,000 images from various public indoor datasets and implement a scene data generation pipeline with a semi-automated technique to establish relationships and estimate distances among indoor objects. By utilizing this enriched data, we conduct various training recipes and finish SceneVLM. Our experiments demonstrate that \rootname facilitates indoor scene understanding and proves effective in diverse downstream applications, such as 3D scene generation and embodied AI. The code will be released at \url{https://github.com/harrytea/ROOT}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。