SWRNet: A Deep Learning Approach for Small Surface Water Area Recognition Onboard Satellite

Code

• github.com/trongan93/swrnet
  Officialpytorch★ 3

Abstract

This article proposes a deep learning approach for small surface water recognition using multispectral satellite imaging, which reduces the computational complexity by 18.66 times and increases the accuracy of surface water recognition by up to 14.1%. The proposed model uses near infrared combined with RGB spectral imagery to increase the accuracy of surface water recognition. In addition, since surface water only accounts for a small percentage of the remote sensing dataset, thus creating an imbalance problem, a proposed loss function is introduced to combine region-based and distribution-based loss. This article introduces an adaptive factor that automatically adjusts the weighting between distribution- and region-based loss functions. The proposed adaptive factor is determined based on the loss value of the previous training step. The mean intersection over union of surface water between predicted and ground truth regions is recorded as 0.80.

Tasks

Reproductions