Semantic Segmentation of High Spatial Resolution Remote Sensing Imagery Based on Weighted Attention U-Net

被引:1
|
作者
Zhang, Yue [1 ]
Wang, Leiguang [2 ]
Yang, Ruiqi [3 ]
Chen, Nan [1 ]
Zhao, Yili [3 ]
Dai, Qinling [4 ]
机构
[1] Southwest Forestry Univ, Fac Forestry, Kunming, Yunnan, Peoples R China
[2] Southwest Forestry Univ, Inst Big Data & Artificial Intelligence, Kunming, Yunnan, Peoples R China
[3] Southwest Forestry Univ, Coll Big Data & Intelligent Engn, Kunming, Yunnan, Peoples R China
[4] Southwest Forestry Univ, Art & Design Coll, Kunming, Yunnan, Peoples R China
关键词
Semantic segmentation; deep learning; attention gate model; weighted attention U-Net; GID dataset;
D O I
10.1117/12.2680206
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, with the development of deep learning and attention mechanism, more research has been carried out to realize semantic image segmentation based on deep learning integrated attention mechanisms. However, the current semantic segmentation methods have low segmentation accuracy, high computation cost, and serious loss of detailed information. In this paper, a lightweight designed attention gate model was introduced to reduce the computation cost. And because it can suppress irrelevant regions in the input image, while highlighting the salient features of specific tasks, the combination of the two weighting factors input features (x(l)) and gating signal (g) in this structure can improve segmentation accuracy and reduce loss of detail. Therefore, this study used the weighted attention U-Net network to perform semantic segmentation on the GID dataset and finally evaluated it on the four indicators of Precision, Recall, F1-Sorce, and mIoU. This result shows that different weight values have a more significant impact on the experimental results. The attention U-Net with the best weight combination compared with the traditional U-Net network, Precision, Recall, F1-Sorce, and mIoU are increased by 0.88%, 1.4%, 1.13%, and 1.2%, respectively. Compared with the original attention U-Net, Precision, Recall, F1-Sorce, and mIoU are increased by 0.86%, 1.24%, 1.04%, and 1.75%, respectively.
引用
收藏
页数:9
相关论文
共 50 条
  • [21] Aircraft segmentation in remote sensing images based on multi-scale residual U-Net with attention
    Wang, Xuqi
    Zhang, Shanwen
    Huang, Lei
    MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (06) : 17855 - 17872
  • [22] An attention-fused network for semantic segmentation of very-high-resolution remote sensing imagery
    Yang, Xuan
    Li, Shanshan
    Chen, Zhengchao
    Chanussot, Jocelyn
    Jia, Xiuping
    Zhang, Bing
    Li, Baipeng
    Chen, Pan
    ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2021, 177 : 238 - 262
  • [23] Automated seismic semantic segmentation using attention U-Net
    Alsalmi, Haifa
    Elsheikh, Ahmed H.
    GEOPHYSICS, 2024, 89 (01) : WA247 - WA263
  • [24] Attention-augmented U-Net (AA-U-Net) for semantic segmentation
    Kumar T. Rajamani
    Priya Rani
    Hanna Siebert
    Rajkumar ElagiriRamalingam
    Mattias P. Heinrich
    Signal, Image and Video Processing, 2023, 17 : 981 - 989
  • [25] Attention-augmented U-Net (AA-U-Net) for semantic segmentation
    Rajamani, Kumar T.
    Rani, Priya
    Siebert, Hanna
    ElagiriRamalingam, Rajkumar
    Heinrich, Mattias P.
    SIGNAL IMAGE AND VIDEO PROCESSING, 2023, 17 (04) : 981 - 989
  • [26] AANet: an attention-based alignment semantic segmentation network for high spatial resolution remote sensing images
    Xue, Gunagkuo
    Liu, Yikun
    Huang, Yuwen
    Li, Mingsong
    Yang, Gongping
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (13) : 4836 - 4852
  • [27] Oceanic Eddy Identification Using Pyramid Split Attention U-Net With Remote Sensing Imagery
    Zhao, Nan
    Huang, Baoxiang
    Yang, Jie
    Radenkovic, Milena
    Chen, Ge
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [28] Research on High Altitude Remote Sensing Building Segmentation Based on Improved U-Net Algorithm
    SHI Mengyuan
    GAO Junchai
    Instrumentation, 2021, 8 (04) : 47 - 54
  • [29] CS U-NET: A Medical Image Segmentation Method Integrating Spatial and Contextual Attention Mechanisms Based on U-NET
    Zhang, Fanyang
    Fan, Zhang
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2025, 35 (02)
  • [30] Convolutional block attention module U-Net: a method to improve attention mechanism and U-Net for remote sensing images
    Zhang, Yanjun
    Kong, Jiayuan
    Long, Sifang
    Zhu, Yuanhao
    He, Fushuai
    JOURNAL OF APPLIED REMOTE SENSING, 2022, 16 (02)