Water body extraction from high spatial resolution remote sensing images based on enhanced U-Net and multi-scale information fusion

Abstract

Employing deep learning techniques for the semantic segmentation of remote sensing images has emerged as a prevalent approach for acquiring information about water bodies. Yet, current models frequently fall short in accurately extracting water bodies from high-resolution remote sensing images, as these images often present intricate details of terrestrial objects and complex backgrounds. Vegetation, shadows, and other objects close to water boundaries have increased similarity to water bodies. Moreover, water bodies in high-resolution images have different boundary complexities, shapes, and sizes. This situation makes it somewhat challenging to accurately distinguish water bodies in high-resolution images. To overcome these difficulties, this paper presents a novel network model named EU-Net, specifically designed to extract water bodies from high-resolution remote sensing images. The proposed EU-Net model, with U-net as the backbone network, incorporates improved residual connections and attention mechanisms, and designs multi-scale dilated convolution and multi-scale feature fusion modules to enhance water body extraction performance in complex scenarios. Specifically, in the proposed model, improved residual connections are introduced to enable the learning of more complex features; the attention mechanism is employed to improve the model's discriminative ability by focusing on important channels and spatial areas. The implemented multi-scale dilated convolution technique enhances the model's receptive field while maintaining the same number of parameters. The designed multi-scale feature fusion module is capable of processing both small-scale details and large-scale structures in images, while simultaneously modeling the spatial context relationships of features at different scales. Experimental results validate the superior performance of EU-Net in accurately identifying water bodies from high-resolution remote sensing images, outperforming current models in terms of water extraction accuracy.