Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 3, 397(2023)

Water body segmentation in remote sensing images based on multi-scale fusion attention module improved UNet

Tian-tian SHI1,2, Zhong-hua GUO1,2、*, Xiang YAN1,2, and Shi-qin WEI1,2
Author Affiliations
  • 1School of Physics and Electronic-Electrical Engineering,Ningxia University,Yinchuan 750021,China
  • 2Ningxia Key Lab on Information Sensing & Intelligent Desert,Ningxia University,Yinchuan 750021,China
  • show less

    A multi-scale fusion attention module improved UNet network is proposed for the water body segmentation task of remote sensing images, A-MSFAM-UNet, which achieves end-to-end high-resolution remote sensing images in the GF-2 remote sensing image water body segmentation task. Firstly, aiming at insensitivity of local information caused by global pooling operation of previous attention module, a multi-scale fusion attention module (MSFAM) is designed, which uses point convolution to fuse channel global information and depthwise separable convolution. The loss of information caused by global pooling is made up. MSFAM is adopted to redistribute the weights of feature points after UNet skip connection to improve the efficiency of feature fusion and enhance network ability to obtain information at different scales. Secondly, the atrous convolution is applied to VGG16 backbone network to expand receptive field and aggregate global information without loss of resolution. The results show that A-MSFAM-UNet outperforms other channel attention (SENet, ECANet) improved UNet, and achieves mean intersection over union(MIoU)、mean pixel accruary(MPA) and accuracy(Acc) of 96.02%, 97.98% and 99.26% on the GF-2 water body segmentation dataset.

    Tools

    Get Citation

    Copy Citation Text

    Tian-tian SHI, Zhong-hua GUO, Xiang YAN, Shi-qin WEI. Water body segmentation in remote sensing images based on multi-scale fusion attention module improved UNet[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(3): 397

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Jul. 7, 2022

    Accepted: --

    Published Online: Apr. 3, 2023

    The Author Email: Zhong-hua GUO (guozhh@nxu.edu.cn)

    DOI:10.37188/CJLCD.2022-0232

    Topics