地球信息科学学报 ›› 2021, Vol. 23 ›› Issue (5): 918-927.doi: 10.12082/dqxxkx.2021.200397

• 遥感科学与应用技术 • 上一篇    下一篇

结合模糊度和形态学指数约束的深度学习建筑物提取

许泽宇1,2(), 沈占锋1,4,*(), 李杨1,3, 柯映明1,4, 李硕1,4, 王浩宇1,2, 焦淑慧1,2   

  1. 1.中国科学院空天信息创新研究院 国家遥感应用工程技术研究中心,北京 100101
    2.中国科学院大学 电子电气与通信工程学院,北京 100049
    3.中国科学院空天信息创新研究院 遥感科学国家重点实验室,北京 100101
    4.中国科学院大学 资源与环境学院,北京 100049
  • 收稿日期:2020-07-26 修回日期:2020-11-23 出版日期:2021-05-25 发布日期:2021-07-25
  • 通讯作者: 沈占锋
  • 作者简介:许泽宇(1995— ),男,河北保定人,硕士生,研究方向为遥感信息提取。E-mail:xuzeyu@aircas.ac.cn
  • 基金资助:
    国家重点研发计划项目(2018YFB0505000);国家重点研发计划项目(2017YFB0504204);国家重点研发计划项目(2016YFC0803109);国家自然科学基金项目(41971375);国家自然科学基金项目(41871283)

Building Extraction by Deep Learning Method Combined with Ambiguity and Morphological Index Constraints

XU Zeyu1,2(), SHEN Zhanfeng1,4,*(), LI Yang1,3, KE Yingming1,4, LI Shuo1,4, WANG Haoyu1,2, JIAO Shuhui1,2   

  1. 1. National Engineering Research Center for Geomatics, Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing 100101, China
    2. School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences, Beijing 100049, China
    3. State Key Laboratory of Remote Sensing Science, Aerospace Information Research Institute, Chinese Academy of Sciences, Beijing 100101, China
    4. College of Resources and Environment, University of Chinese Academy of Sciences, Beijing 100049, China
  • Received:2020-07-26 Revised:2020-11-23 Online:2021-05-25 Published:2021-07-25
  • Contact: SHEN Zhanfeng
  • Supported by:
    National Key Research and Development Project(2018YFB0505000);National Key Research and Development Project(2017YFB0504204);National Key Research and Development Project(2016YFC0803109);National Natural Science Foundation of China(41971375);National Natural Science Foundation of China(41871283)

摘要:

基于高分辨率遥感影像的建筑物提取一直是研究的热点问题,深度学习的深层次特征提取方法,非常适合高分辨率影像中建筑物的提取,但使用深度学习提取建筑物时,大多以改变网络结构为主进行算法优化,很少与其他方法结合。本文研究在改进深度学习网络结构的基础上,结合影像模糊度约束增强、形态学建筑指数约束增强等方法,对建筑物提取方法进行更全面更有针对性的改进。本文主要改进内容为:① 提出PwDeepLab网络,该网络基于DeepLab v3+网络结构,在特征融合方式和损失函数等方面进行了改进。② 提出模糊度约束方法,在固定影像块大小的情况下,通过影像模糊度约束对影像进行上采样增强。③ 提出形态学指数约束方法,通过形态学建筑物指数(MBI)约束范围拉伸增强的方法,在较少改变原始影像特征的情况下,突出建筑信息。本文在Massachusetts数据集和武汉大学的Satellite Dataset Ⅱ(East Asia) 数据集上进行验证, 2个数据集的主要建筑类型存在较大区别。本文提出的方法在2个数据集上的精度相对于DeepLab v3+分别提高了10.9%和3.8%,相对于U-Net分别提高了10.0%和9.6%。实验结果表明本文提出的方法对建筑物提取效果有明显提升,且具有很好的鲁棒性和通用性。

关键词: 深度学习, 建筑提取, 高分辨率影像, PwDeepLab模型, 特征融合, 模糊度, 形态学指数, 窗口增强

Abstract:

Extraction of buildings from high-resolution remote sensing images has been a hot topic. It is important to quickly and accurately extract the location and outline of buildings from high-resolution remote sensing images for earthquake disaster assessment, urban and rural planning management, smart city construction, and other fields. However, it is challenging to extract buildings accurately from high-resolution remote sensing images due to the complexity of ground features and the occlusion phenomenon. The classic building extraction algorithms usually have incomplete results with many wrong detections and missed detections. The deep-level feature extraction method of deep learning is very suitable for the extraction of buildings from high-resolution images. However, for the existing researches using deep learning algorithms, most algorithms are optimized by changing the network structure and are rarely combined with other methods. The pertinence of building extraction is not strong. This paper not only studies the influence of the internal structure of the convolutional neural network in deep learning on the extraction results, but also studies the combination of ambiguity, Morphological Building Index (MBI) with deep learning. The main improvements in this paper are as follows: (1) Based on the DeepLab v3+ network structure, we propose the PwDeepLab network which improves feature fusion methods and loss functions; (2) We propose a blur degree constraint method. We define a new blur degree formula to evaluate the image blur degree. In the case of a fixed image block size, the image is upsampled and enhanced by the image blur degree constraint; (3) We propose a morphological index constraint method. The pixels of the original image are stretched and enhanced where the Morphological Building Index (MBI) is above the threshold. Therefore, the building information can be highlighted with less changes to the original image characteristics. Our method is verified using the Massachusetts dataset and the satellite dataset II (East Asia) of Wuhan University. The main building types of the two datasets are quite different. As a result, the accuracy of the proposed method on the two datasets increases by 10.9% and 3.8%, respectively, compared with DeepLab v3+, and increases by 10.0% and 9.6%, respectively, compared with U-Net. The higher accuracy reflects the superiority and robustness of our method. Moreover, the extracted results match the real labels very well in details. The experimental results show that the method proposed in this paper can significantly improve the building extraction results.

Key words: deep learning, building extraction, high resolution image, PwDeepLab, feature fusion, ambiguity, morphological index, window enhancement