Laser & Optoelectronics Progress, Volume. 57, Issue 4, 041015(2020)

Remote Sensing Image Segmentation Model Based on Attention Mechanism

Hang Liu and Xili Wang*
Author Affiliations
  • School of Computer Science, Shaanxi Normal University, Xi'an, Shaanxi 710119, China
  • show less
    Figures & Tables(15)
    Comparison between RSANet and LinkNet. (a) RSANet; (b) LinkNet
    Structure of encoder block
    Structure of decoder block
    Position attention module
    Channel attention module
    Remote sensing images and road labels in Massachusetts Roads dataset. (a) Part of the remote sensing images in the training set; (b) road labels corresponding to images in the training set; (c) some remote sensing images in the test set; (d) road labels corresponding to images in the test set
    Remote sensing images and road labels in DeepGlobe dataset. (a) Part of the remote sensing images in the training set; (b) road labels corresponding to images in the training set; (c) some remote sensing images in the test set; (d) road labels corresponding to images in the test set
    Segmentation results of different depth models in the Massachusetts Roads test set. (a) Original images; (b) labels; (c) segmentation results of LinkNet; (d) segmentation results of RSANet
    Segmentation results of different depth models on the DeepGlobe Road Extraction test set. (a) Original images; (b) labels; (c) segmentation results of LinkNet; (d) segmentation results of RSANet
    • Table 1. Segmentation results of depth models on each test image in Fig. 8

      View table

      Table 1. Segmentation results of depth models on each test image in Fig. 8

      ModelImage 1Image 2Image 3Image 4Image 5
      PrF1-scorePrF1-scorePrF1-scorePrF1-scorePrF1-score
      LinkNet[3]0.8390.8590.9130.8810.8210.8320.8000.8170.7470.779
      RSANet0.9790.9760.9230.8920.8290.8350.8220.8890.8180.827
    • Table 2. Segmentation results of different depth models on the Massachusetts Roads test set

      View table

      Table 2. Segmentation results of different depth models on the Massachusetts Roads test set

      ModelPrF1-scoreTraining time /hInference time per picture /s
      LinkNet[3](baseline)0.8020.815310.31
      LinkNet-PAM0.8120.823320.32
      LinkNet-CAM0.8180.828400.40
      RSANet0.8270.843410.41
    • Table 3. Segmentation results of different depth models on each test image in Fig. 9

      View table

      Table 3. Segmentation results of different depth models on each test image in Fig. 9

      ModelImage 6Image 7Image 8Image 9Image 10
      PrPIOUPrPIOUPrPIOUPrPIOUPrPIOU
      LinkNet[3]0.7700.7120.5970.5150.7330.6310.8860.7220.7290.704
      RSANet0.7920.7350.8240.6960.8350.6940.9070.7900.7730.723
    • Table 4. Segmentation results of different depth models on the DeepGlobe Road Extraction test set

      View table

      Table 4. Segmentation results of different depth models on the DeepGlobe Road Extraction test set

      ModelPPPIOUTraining time /hInference time per picture /s
      LinkNet[3](baseline)0.7850.598310.31
      LinkNet-PAM0.7910.610320.32
      LinkNet-CAM0.7960.616400.40
      RSANet0.8110.624410.41
    • Table 5. Segmentation results of different depth models on the Massachusetts Roads test set

      View table

      Table 5. Segmentation results of different depth models on the Massachusetts Roads test set

      ModelPrPPF1-score
      FCN-4s[2]0.6600.7100.684
      SegNet[18]0.7650.7730.768
      ELU-SegNet[18]0.7730.8520.788
      ELU-SegNet-R[18]0.7800.8470.812
      DCED[19]0.8390.8250.829
      RSANet0.8270.8590.843
    • Table 6. Segmentation results of different depth models on the DeepGlobe Road Extraction test set

      View table

      Table 6. Segmentation results of different depth models on the DeepGlobe Road Extraction test set

      ModelPPF1-scoreTraining time /hInference time per picture /s
      U-Net[1]0.7640.597700.94
      SegNet[20]0.7740.602981.8
      LinkNet[3](baseline)0.7850.598310.31
      RSANet0.8110.624410.41
    Tools

    Get Citation

    Copy Citation Text

    Hang Liu, Xili Wang. Remote Sensing Image Segmentation Model Based on Attention Mechanism[J]. Laser & Optoelectronics Progress, 2020, 57(4): 041015

    Download Citation

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

    Category: Image Processing

    Received: Jun. 22, 2019

    Accepted: Aug. 14, 2019

    Published Online: Feb. 20, 2020

    The Author Email: Wang Xili (wangxili@snnu.edu.cn)

    DOI:10.3788/LOP57.041015

    Topics