Improvement and application of UNet network for avoiding the effect of urban dense high-rise buildings and other feature shadows on water body extraction

Abstract

Finding a means to extract water body information efficiently and accurately from high-resolution remote sensing images has been an important research direction in the field of water body extraction in recent years. However, shadows from buildings and other obstacles interfere with the accuracy of water body extraction. To address this problem, this paper proposes a neural network method incorporating an attention mechanism for water body extraction. This paper is based on the U-Net convolutional neural network and adds the squeeze-and-excitation module of SENet, an attention mechanism, to the downsampling process of the U-Net network. The module weights the feature maps so that the network focuses more on the features of the water body information and thus reduces attention to the shadow features from buildings and other features, thus improving the accuracy of image segmentation. The dropout and batch normalization layers are also added to improve the generalization ability and stability of the model. In this paper, a water extraction network SE-CU-Net model is presented to overcome the shadowing effect from buildings and other features. Using GF-2 images of Jiangsu province as the data source, the recognition results of this paper are compared with Dense-Net, Res-Net, Seg-Net, U-net, SVM, and RF. Through the comparison experiments, the model of this paper can not only better overcome the influence of shadows from buildings and other features, but it also has a stronger recognition ability and recognition effect. The average ASCR, Precision, mIoU, OA, F1-Score and kappa coefficients in the three tested areas reached 98.27%, 97.17%, 89.33%, 98.2%, 89.3% and 0.883, respectively, with significantly higher accuracy than the other six classical methods, verifying the effectiveness of the model in overcoming the influence of shadows from buildings and other features in water body extraction research.