arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2512.02413cs.CVcs.AI

增强平面识别:一种混合Mix-Transformer和U-Net方法用于精确墙体分割

Enhancing Floor Plan Recognition: A Hybrid Mix-Transformer and U-Net Approach for Precise Wall Segmentation

Tyumen State University(秋明国立大学)

Dmitriy Parashchuk, Alexey Kaspshitskiy, Yuriy Karyakin

AI总结:

本文提出MitUNet,结合Mix-Transformer和U-Net,通过改进的损失函数提升墙体分割精度,实验显示其在CubiCasa5k数据集上表现优异,为自动化3D重建提供基础。

摘要:

自动从2D平面图重建室内空间的3D模型需要高精度的结构元素语义分割,特别是墙体。然而,现有方法常难以检测薄结构并保持几何精度。为此,我们引入MitUNet,一种混合神经网络,旨在弥合全局语义上下文与细粒度结构细节之间的差距。我们的架构结合了Mix-Transformer编码器和带有空间和通道注意力块的U-Net解码器。通过Tversky损失函数优化,该方法在精度和召回率之间取得平衡,确保准确的边界恢复。在CubiCasa5k数据集和区域数据集上的实验表明,MitUNet在生成结构正确且边界准确的掩码方面优于标准模型。此工具为自动化3D重建流水线提供了稳健的基础。为确保可重复性和促进未来研究,源代码和区域数据集已公开在https://github.com/aliasstudio/mitunet和https://doi.org/10.5281/zenodo.17871079。

英文摘要:

Automatic 3D reconstruction of indoor spaces from 2D floor plans necessitates high-precision semantic segmentation of structural elements, particularly walls. However, existing methods often struggle with detecting thin structures and maintaining geometric precision. To address this, we introduce MitUNet, a hybrid neural network designed to bridge the gap between global semantic context and fine-grained structural details. Our architecture combines a Mix-Transformer encoder with a U-Net decoder enhanced with spatial and channel attention blocks. Optimized with the Tversky loss function, this approach achieves a balance between precision and recall, ensuring accurate boundary recovery. Experiments on the CubiCasa5k dataset and the regional dataset demonstrate MitUNet's superiority in generating structurally correct masks with high boundary accuracy, outperforming standard models. This tool provides a robust foundation for automated 3D reconstruction pipelines. To ensure reproducibility and facilitate future research, the source code and the regional dataset are publicly available at https://github.com/aliasstudio/mitunet and https://doi.org/10.5281/zenodo.17871079, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑