Bulletin of Surveying and Mapping ›› 2024, Vol. 0 ›› Issue (8): 26-30.doi: 10.13474/j.cnki.11-2246.2024.0805

Previous Articles    

Remote sensing image water body extraction based on U-Net, U-Net++ and Attention-U-Net networks

LI Zhenxuan, HUANG Miner, GAO Fei, TAO Tingye, WU Zhaofu, ZHU Yongchao   

  1. School of Civil and Hydraulic Engineering, Hefei University of Technology, Hefei 230009, China
  • Received:2024-04-01 Published:2024-09-03

Abstract: Currently, the application of deep learning in the extraction of water bodies from high-resolution remote sensing images has become a research hotspot in the remote sensing field. Among them, algorithms based on the U-Net network have demonstrated good performance in water body extraction. However, there is scarce research that provides in-depth and detailed comparisons of the performance differences of different U-Net network algorithms in water body extraction tasks. Therefore, this article selects three convolutional neural networks, named U-Net, U-Net++, and Attention-U-Net, and based on the GID dataset, draws conclusions through experiments and quantitative analysis. The results indicate that: U-Net++ achieves the highest training accuracy, followed by U-Net and Attention-U-Net, with accuracies of 0.912, 0.907, and 0.899 respectively. U-Net++ exhibits superior edge extraction capability compared to the other two networks. In segmenting different types of water bodies and distinguishing non-water areas similar to water bodies in remote sensing images, U-Net++ shows significantly better extraction results, while U-Net and Attention-U-Net are prone to omission errors and exhibit suboptimal performance.

Key words: water body extraction, high-resolution remote sensing imagery, U-Net

CLC Number: