3D-Front
收藏资源简介:
3D-Front数据集是由上海人工智能实验室、香港大学和香港中文大学共同创建的,包含约6500个室内场景的2D顶视图布局,用于训练和评估3D室内场景合成模型。数据集从ATISS数据集中筛选而来,并通过GPT-4o模型生成CoT注释,以帮助模型理解和组织空间信息,提高空间推理能力。3D-Front数据集的创建旨在解决现有3D室内布局数据集规模、多样性和真实感不足的问题,以促进3D室内场景合成的进一步研究。
The 3D-Front Dataset was jointly created by Shanghai AI Laboratory, The University of Hong Kong, and The Chinese University of Hong Kong. It contains approximately 6,500 2D top-view layouts of indoor scenes, which are used for training and evaluating 3D indoor scene synthesis models. This dataset is screened from the ATISS dataset, and CoT annotations are generated using the GPT-4o model to assist models in understanding and organizing spatial information and enhancing their spatial reasoning capabilities. The creation of the 3D-Front Dataset aims to address the shortcomings of existing 3D indoor layout datasets in terms of scale, diversity and realism, so as to advance further research on 3D indoor scene synthesis.
Direct Numerical Layout Generation for 3D Indoor Scene Synthesis via Spatial Reasoning
作者信息
- Xingjian Ran (上海人工智能实验室, 香港大学)
- Yixuan Li (香港中文大学)
- Linning Xu (香港中文大学)
- Mulin Yu (上海人工智能实验室)
- Bo Dai (香港大学)
生成样本示例
粗粒度提示
- 计算机教室:配备独立电脑桌、人体工学椅、投影屏幕、桌子、打印机和网络集线器
- 走廊:包含鞋架、衣帽钩、镜子、长凳、伞架、桌子和地毯
- 厨房:配备冰箱、炉灶、水槽、台面、餐桌、多个橱柜和几个吧凳
中粒度提示
- 家庭影院:大屏幕靠墙放置,两侧有扬声器,前方圆形咖啡桌摆放爆米花和薯片,后方沙发周围布置豆袋椅
- 车库:中央停放汽车,后墙有储物架,钉板下放置工作台和工具箱,附近摆放自行车和割草机
- 教室:前方教师桌面向学生桌椅排列,前墙安装白板,天花板悬挂投影仪,侧面放置高书架
细粒度提示
- 游戏室:中央放置桌上足球台,两侧布置游戏椅,墙边木制支架上放置大电视,电视左侧有银色迷你冰箱,另一面墙的书架装满桌游,墙上展示两张海报
- 酒窖:两面墙放置木质酒架存放酒瓶,另一面墙堆叠橡木桶和木箱,中央放置品酒桌配单椅,桌上摆放两个酒杯和开瓶器,桶上方墙面安装温度计,天花板有两个圆形灯具
- 工作台:左前方放置笔记本电脑,右侧有彩色铅笔玻璃杯,中央后方摆放蓝色花瓶插花,花瓶前放置黑框眼镜,左侧有白色咖啡杯,桌前配置木椅
方法概述
-
训练阶段:
- BEV布局生成器基于3D-Front数据集微调,采用GPT-4o生成的CoT注释指导
- 通过DPO优化,利用空间评估器(VLM)和定量评估器(推理LLM)产生的CoT-Grounded生成布局奖励
-
推理阶段:
- 文本提示输入BEV布局生成器产生2D布局
- 3D布局生成器将2D布局提升为3D
- 通过迭代资产-布局对齐机制,使用空间和定量评估器提供反馈,确保布局与对象生成器产生的3D资产一致性
核心贡献
- 直接生成数值3D布局,无需中间表示和约束优化
- 三阶段生成流程:BEV布局生成→3D空间提升→物体放置细化
- 基于3D-Front数据集采用CoT激活实现显式空间推理
- 设计CoT-Grounded生成布局奖励增强泛化能力和空间规划
- 推理阶段通过上下文学习解决资产-布局失配问题

- 1Direct Numerical Layout Generation for 3D Indoor Scene Synthesis via Spatial Reasoning上海人工智能实验室, 香港大学, 香港中文大学 · 2025年



