Water extraction in SAR images using features analysis and dual-threshold graph cut model

Abstract

Timely identifying and detecting water bodies from SAR images are significant for flood monitoring and water resources management. In recent decades, deep learning has been applied to water extraction but is subject to the large difficulty of acquiring SAR dataset of various water bodies types, as well as heavy labeling work. In addition, the traditional methods mostly occur over the large, open lakes and rivers, rarely focusing on complex areas such as the urban water, and cannot automatically acquire the classification threshold. To address these issues, a novel water extraction method is proposed with high accuracy in this paper. Firstly, a multiscale feature extraction using a Gabor filter is conducted to reduce the noise and roughly identify water feature. Secondly, we apply the Otsu algorithm as well as a voting strategy to initially extract the homogeneous regions and for subsequent Gaussian mixture model (GMM). Finally, the dual threshold is obtained from the fitted Gaussian distribution of water and non-water, which is integrated into the graph cut model to redefine the weights of the edges, then constructing the energy function of the water map. The dual-threshold graph cut (DTGC) model precisely pinpoints the water location by minimizing the energy function. To verify the efficiency and robustness, our method and comparison methods, including the IGC method and IACM method, are tested on six different types of water bodies, by performing the accuracy assessment via comparing outcomes with the manually labeled ground truth. The qualitative and quantitative results show that the overall accuracy of our method for the whole dataset all surpasses 99%, along with an obvious improvement of the Kappa, F1-score, and IoU indicators. Therefore, DTGC method has the absolute advantage of automatically capturing water maps in different scenes of SAR images without specific prior knowledge and can also determine the optimal threshold range.