arXiv:2509.15459cs.CVcs.AI2025-09NeurIPS被引 7

用连续边结构重建楼图,抗噪强且细节准。

CAGE: Continuity-Aware edGE Network Unlocks Robust Floorplan Reconstruction

  • 以有向连续边为基本单元,避免碎片化布局。
  • 在两个数据集上房间F1达99.1%,角点89.3%。
  • 适合需要高精度建筑平面的场景重建任务。

我们提出CAGE(Continuity-Aware edGE)网络,一种直接从点云密度图重建矢量楼图的鲁棒框架。传统基于角点的多边形表示对噪声和不完整观测敏感,常导致布局断裂或不合理。近期线段分组方法虽利用结构线索提升鲁棒性,但仍难恢复精细几何细节。为此,我们采用原生边中心的建模方式,将每段墙视为有向、几何连续的边,从而推断出连贯的楼图结构,确保无孔洞、拓扑正确的房间边界,同时增强鲁棒性并减少伪影。为此,设计了双查询变换器解码器,在去噪框架中融合扰动与潜在查询,不仅稳定优化过程,还加速收敛。在Structured3D和SceneCAD上的大量实验表明,CAGE达到当前最优性能:房间F1为99.1%,角点91.7%,角度89.3%。方法还展现出强跨数据集泛化能力,验证了架构创新的有效性。代码与预训练模型见项目页:https://github.com/ee-Liu/CAGE.git。

原文摘要 · Abstract (English)

We present CAGE (Continuity-Aware edGE) network, a robust framework for reconstructing vector floorplans directly from point-cloud density maps. Traditional corner-based polygon representations are highly sensitive to noise and incomplete observations, often resulting in fragmented or implausible layouts.Recent line grouping methods leverage structural cues to improve robustness but still struggle to recover fine geometric details. To address these limitations,we propose a native edge-centric formulation, modeling each wall segment as a directed, geometrically continuous edge. This representation enables inference of coherent floorplan structures, ensuring watertight, topologically valid room boundaries while improving robustness and reducing artifacts. Towards this design, we develop a dual-query transformer decoder that integrates perturbed and latent queries within a denoising framework, which not only stabilizes optimization but also accelerates convergence. Extensive experiments on Structured3D and SceneCAD show that CAGE achieves state-of-the-art performance, with F1 scores of 99.1% (rooms), 91.7% (corners), and 89.3% (angles). The method also demonstrates strong cross-dataset generalization, underscoring the efficacy of our architectural innovations. Code and pretrained models are available on our project page: https://github.com/ee-Liu/CAGE.git.

楼图重建矢量生成连续边去噪框架

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