#布局预测

HorizonNet项目通过1D表示和全景拉伸数据增强实现了房间布局学习，能够生成立方体或一般形状的房间布局。该项目提供纯Python实现，支持图像推断、3D布局查看、姿态校正、全景拉伸增强、2D和3D布局定量评估以及数据集准备和训练。支持PanoContext、Stanford2D3D、Structured3D和Zillow Indoor等数据集，提供预训练模型。详细的安装、推断和可视化教程使用户能够高效地使用该工具。