Offset-Guided Attention Network for Room-Level Aware Floor Plan Segmentation
Zhangyu Wang, Ningyuan Sun
Abstract
Recognition of floor plans has been a challenging and popular task. However, existing approaches are struggling to make accurate room-level unified predictions, seriously limiting their visual quality and applicability. In this paper, we propose a novel approach to recognize the floor plan layouts with a newly proposed Offset-Guided Attention mechanism to improve the semantic consistency within a room. In addition, we present a Feature Fusion Attention module that leverages the channel-wise attention to encourage the consistency of the room, wall, and door predictions, further enhancing the room-level semantic consistency. Experimental results manifest our approach is able to improve room-level semantic consistency and outperforms the existing works both qualitatively and quantitatively. To summarize, this paper tackles inconsistent semantic prediction issues in floor plan segmentation and enhances visual plausibility, quantitative performance, and practical applicability.