论文标题

BuildMapper:一个完全可学习的矢量化建筑轮廓提取框架

BuildMapper: A Fully Learnable Framework for Vectorized Building Contour Extraction

论文作者

Wei, Shiqing, Zhang, Tao, Ji, Shunping, Luo, Muying, Gong, Jianya

论文摘要

基于深度学习的方法显着增强了从遥感图像中自动构建提取的研究。但是,由于方法的难度,建筑结构的多样性和不完美的成像条件,划定像人类一样的矢量化和常规建筑轮廓仍然非常具有挑战性。在本文中,我们提出了一个名为BuildMapper的第一个端到端可学习的建筑轮廓提取框架,该框架可以像人类一样直接有效地描绘建筑物多边形。 BuildMapper由两个主要组成部分组成:1)生成初始建筑轮廓的轮廓初始化模块; 2)轮廓演变模块,既执行轮廓顶点变形又减少,从而消除了现有方法中使用的复杂经验后处理的需求。在这两个组件中,我们都提供了新的想法,包括一种可学习的轮廓初始化方法,用于替换经验方法,动态预测和地面真相顶点对静态顶点对应问题问题,以及用于顶点信息提取和聚集的轻量级编码器,这些编码器受益于一般轮廓方法;以及一个精心设计的顶点分类头,用于建筑物角顶点检测,该灯在直接的结构化建筑物轮廓提取上投射了灯。我们还构建了合适的大型建筑数据集WHU-MIX(Vector)建筑数据集,以使基于轮廓的建筑提取方法的研究受益。在WHU-MIX(Vector)数据集,WHU数据集和Crowdai数据集上进行的广泛实验证明了BuildMapper可以实现最先进的性能,而基于段的基于段的方法和基于轮廓的方法具有更高的掩码平均精度(AP)和边界AP。

Deep learning based methods have significantly boosted the study of automatic building extraction from remote sensing images. However, delineating vectorized and regular building contours like a human does remains very challenging, due to the difficulty of the methodology, the diversity of building structures, and the imperfect imaging conditions. In this paper, we propose the first end-to-end learnable building contour extraction framework, named BuildMapper, which can directly and efficiently delineate building polygons just as a human does. BuildMapper consists of two main components: 1) a contour initialization module that generates initial building contours; and 2) a contour evolution module that performs both contour vertex deformation and reduction, which removes the need for complex empirical post-processing used in existing methods. In both components, we provide new ideas, including a learnable contour initialization method to replace the empirical methods, dynamic predicted and ground truth vertex pairing for the static vertex correspondence problem, and a lightweight encoder for vertex information extraction and aggregation, which benefit a general contour-based method; and a well-designed vertex classification head for building corner vertices detection, which casts light on direct structured building contour extraction. We also built a suitable large-scale building dataset, the WHU-Mix (vector) building dataset, to benefit the study of contour-based building extraction methods. The extensive experiments conducted on the WHU-Mix (vector) dataset, the WHU dataset, and the CrowdAI dataset verified that BuildMapper can achieve a state-of-the-art performance, with a higher mask average precision (AP) and boundary AP than both segmentation-based and contour-based methods.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源