PolyRoom: Room-aware Transformer for Floorplan Reconstruction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Yuzhou, Zhu, Lingjie, Ma, Xiaodong, Ye, Hanqiao, Gao, Xiang, Zheng, Xianwei, Shen, Shuhan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929420621578240
author Liu, Yuzhou
Zhu, Lingjie
Ma, Xiaodong
Ye, Hanqiao
Gao, Xiang
Zheng, Xianwei
Shen, Shuhan
author_facet Liu, Yuzhou
Zhu, Lingjie
Ma, Xiaodong
Ye, Hanqiao
Gao, Xiang
Zheng, Xianwei
Shen, Shuhan
contents Reconstructing geometry and topology structures from raw unstructured data has always been an important research topic in indoor mapping research. In this paper, we aim to reconstruct the floorplan with a vectorized representation from point clouds. Despite significant advancements achieved in recent years, current methods still encounter several challenges, such as missing corners or edges, inaccuracies in corner positions or angles, self-intersecting or overlapping polygons, and potentially implausible topology. To tackle these challenges, we present PolyRoom, a room-aware Transformer that leverages uniform sampling representation, room-aware query initialization, and room-aware self-attention for floorplan reconstruction. Specifically, we adopt a uniform sampling floorplan representation to enable dense supervision during training and effective utilization of angle information. Additionally, we propose a room-aware query initialization scheme to prevent non-polygonal sequences and introduce room-aware self-attention to enhance memory efficiency and model performance. Experimental results on two widely used datasets demonstrate that PolyRoom surpasses current state-of-the-art methods both quantitatively and qualitatively. Our code is available at: https://github.com/3dv-casia/PolyRoom/.
format Preprint
id arxiv_https___arxiv_org_abs_2407_10439
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PolyRoom: Room-aware Transformer for Floorplan Reconstruction
Liu, Yuzhou
Zhu, Lingjie
Ma, Xiaodong
Ye, Hanqiao
Gao, Xiang
Zheng, Xianwei
Shen, Shuhan
Computer Vision and Pattern Recognition
Reconstructing geometry and topology structures from raw unstructured data has always been an important research topic in indoor mapping research. In this paper, we aim to reconstruct the floorplan with a vectorized representation from point clouds. Despite significant advancements achieved in recent years, current methods still encounter several challenges, such as missing corners or edges, inaccuracies in corner positions or angles, self-intersecting or overlapping polygons, and potentially implausible topology. To tackle these challenges, we present PolyRoom, a room-aware Transformer that leverages uniform sampling representation, room-aware query initialization, and room-aware self-attention for floorplan reconstruction. Specifically, we adopt a uniform sampling floorplan representation to enable dense supervision during training and effective utilization of angle information. Additionally, we propose a room-aware query initialization scheme to prevent non-polygonal sequences and introduce room-aware self-attention to enhance memory efficiency and model performance. Experimental results on two widely used datasets demonstrate that PolyRoom surpasses current state-of-the-art methods both quantitatively and qualitatively. Our code is available at: https://github.com/3dv-casia/PolyRoom/.
title PolyRoom: Room-aware Transformer for Floorplan Reconstruction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.10439