SAR image water extraction using the attention U-net and multi-scale level set method: flood monitoring in South China in 2020 as a test case

Abstract

Level set method has been extensively used for image segmentation, which is a key technology of water extraction. However, one of the problems of the level-set method is how to find the appropriate initial surface parameters, which will affect the accuracy and speed of level set evolution. Recently, the semantic segmentation based on deep learning has opened the exciting research possibilities. In addition, the Convolutional Neural Network (CNN) has shown a strong feature representation capability. Therefore, in this paper, the CNN method is used to obtain the initial SAR image segmentation map to provide deep a priori information for the zero-level set curve, which only needs to describe the general outline of the water body, rather than the accurate edges. Compared with the traditional circular and rectangular zero-level set initialization method, this method can converge to the edge of the water body faster and more precisely; it will not fall into the local minimum value and be able to obtain accurate segmentation results. The effectiveness of the proposed method is demonstrated by the experimental results of flood disaster monitoring in South China in 2020.

Keywords

• water extraction
• flood monitoring
• level set
• attention u-net
• convolutional neural network (cnn)