1. HA U-Net: Improved Model for Building Extraction From High Resolution Remote Sensing Imagery
- Author
-
Leilei Xu, Yujun Liu, Peng Yang, Hao Chen, Hanyue Zhang, Dan Wang, and Xin Zhang
- Subjects
Deep learning ,building extraction ,holistically-nested neural network ,attention mechanism ,weight mapping ,watershed algorithm ,Electrical engineering. Electronics. Nuclear engineering ,TK1-9971 - Abstract
Automatic extraction of buildings from high-resolution remote sensing images becomes an important research. Since the convolutional neural network can perform pixel-level segmentation, this technology has been applied in this field. But the increase in resolution prone to blurry segmentation because the model needs more edge detail and multi-scale detail learning. To solve this problem, a method is proposed in this paper, which consists of three parts: (1) an improved model named Holistically-Nested Attention U-Net (HA U-Net) is designed, which integrates the attention mechanism and multi-scale nested modules to supervise prediction; (2) During model training, an improved weighted loss function is proposed to make the designed model more focused on learning boundary features; (3) watershed algorithm is exploited for image post-processing to optimize segmentation results. The designed HA U-Net performs well on WHU Building Dataset and Urban3d Challenge dataset, and achieves 9.31%, 2.17% better F1-score and 10.78%, 1.77% better IOU than the standard U-Net respectively. The experimental results indicate that the proposed method can well solve the building adhesion problem. The research can serve as updating geographic databases.
- Published
- 2021
- Full Text
- View/download PDF