arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.09643cs.CV
收藏

HiT:基于分层Transformer的建筑制图

HiT: Building Mapping with Hierarchical Transformers

State Key Laboratory of Virtual Reality Technology and Systems, Beihang University(北京航空航天大学虚拟现实技术与系统国家重点实验室);Hangzhou Innovation Institute, Beihang University(北京航空航天大学杭州创新研究院)

Mingming Zhang, Qingjie Liu, Yunhong Wang

AI总结:

本文提出HiT方法,基于分层Transformer和两阶段检测架构,通过并行多边形头端到端地预测双向序列化顶点,实现高分辨率遥感影像的高质量建筑多边形制图。

摘要:

近年来,基于深度学习的高分辨率遥感影像自动建筑制图方法已被广泛探索。虽然大多数建筑制图模型为地理和制图系统生成建筑的矢量多边形,但主流方法通常将多边形建筑提取分解为多个子问题,包括分割、多边形化和正则化,导致推理过程复杂、精度低且泛化性差。本文提出了一种简单新颖的基于分层Transformer的建筑制图方法,称为HiT,旨在提升高分辨率遥感影像的多边形建筑制图质量。HiT建立在两阶段检测架构之上,通过添加一个与分类和边界框回归头并行的多边形头来实现。HiT同时输出建筑边界框和矢量多边形,且完全端到端可训练。该多边形头将建筑多边形表示为具有双向特征的序列化顶点,这是一种简单而优雅的多边形表示方式,避免了起始或终止顶点的假设。在此新视角下,多边形头采用Transformer编码器-解码器架构,在设计的双向多边形损失监督下预测序列化顶点。此外,多边形头的编码器中引入了结合卷积操作的分层注意力机制,在顶点和边缘层面提供更多建筑多边形的几何结构信息。在两个基准(CrowdAI和Inria数据集)上的综合实验表明,与最先进方法相比,本方法在实例分割和多边形指标上均达到了新的最优水平。定性结果进一步验证了本模型在复杂场景下的优越性和有效性。

英文摘要:

Deep learning-based methods have been extensively explored for automatic building mapping from high-resolution remote sensing images over recent years. While most building mapping models produce vector polygons of buildings for geographic and mapping systems, dominant methods typically decompose polygonal building extraction in some sub-problems, including segmentation, polygonization, and regularization, leading to complex inference procedures, low accuracy, and poor generalization. In this paper, we propose a simple and novel building mapping method with Hierarchical Transformers, called HiT, improving polygonal building mapping quality from high-resolution remote sensing images. HiT builds on a two-stage detection architecture by adding a polygon head parallel to classification and bounding box regression heads. HiT simultaneously outputs building bounding boxes and vector polygons, which is fully end-to-end trainable. The polygon head formulates a building polygon as serialized vertices with the bidirectional characteristic, a simple and elegant polygon representation avoiding the start or end vertex hypothesis. Under this new perspective, the polygon head adopts a transformer encoder-decoder architecture to predict serialized vertices supervised by the designed bidirectional polygon loss. Furthermore, a hierarchical attention mechanism combined with convolution operation is introduced in the encoder of the polygon head, providing more geometric structures of building polygons at vertex and edge levels. Comprehensive experiments on two benchmarks (the CrowdAI and Inria datasets) demonstrate that our method achieves a new state-of-the-art in terms of instance segmentation and polygonal metrics compared with state-of-the-art methods. Moreover, qualitative results verify the superiority and effectiveness of our model under complex scenes.

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

↑