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 条
  • [31] Multiscale feature U-Net for remote sensing image segmentation
    Wei, Youhua
    Liu, Xuzhi
    Lei, Jingxiong
    Yue, Ruihan
    Feng, Jun
    JOURNAL OF APPLIED REMOTE SENSING, 2022, 16 (01)
  • [32] Accurate segmentation of remote sensing cages based on U-Net and voting mechanism
    Chuang, Yu
    Liu Yunpeng
    Hu Zhuhua
    Xin, Xia
    SEVENTH ASIA PACIFIC CONFERENCE ON OPTICS MANUFACTURE (APCOM 2021), 2022, 12166
  • [33] Cloud detection of high-resolution remote sensing image based on improved U-Net
    MeiJie Yin
    Peng Wang
    WeiLong Hao
    Cui Ni
    Multimedia Tools and Applications, 2023, 82 : 25271 - 25288
  • [34] Cloud detection of high-resolution remote sensing image based on improved U-Net
    Yin, MeiJie
    Wang, Peng
    Hao, WeiLong
    Ni, Cui
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (16) : 25271 - 25288
  • [35] Research on Object Detection Technique in High Resolution Remote Sensing Images Based on U-Net
    Wu Zhihuan
    Gao Yongming
    Li Lei
    Fan Junliang
    PROCEEDINGS OF THE 30TH CHINESE CONTROL AND DECISION CONFERENCE (2018 CCDC), 2018, : 2849 - 2853
  • [36] A Siamese Network Based U-Net for Change Detection in High Resolution Remote Sensing Images
    Chen, Tao
    Lu, Zhiyuan
    Yang, Yue
    Zhang, Yuxiang
    Du, Bo
    Plaza, Antonio
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 2357 - 2369
  • [37] Semantic Segmentation of Tumors in Kidneys using Attention U-Net Models
    Geethanjali, T. M.
    Minavathi
    Dinesh, M. S.
    2021 5TH INTERNATIONAL CONFERENCE ON ELECTRICAL, ELECTRONICS, COMMUNICATION, COMPUTER TECHNOLOGIES AND OPTIMIZATION TECHNIQUES (ICEECCOT), 2021, : 286 - 290
  • [38] SEMANTIC SEGMENTATION OF HIGH RESOLUTION REMOTE SENSING IMAGE BASED ON BATCH-ATTENTION MECHANISM
    Su, Yanzhou
    Wu, Yongjian
    Wang, Min
    Wang, Feng
    Cheng, Jian
    2019 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2019), 2019, : 3856 - 3859
  • [39] SMAF-Net: Sharing Multiscale Adversarial Feature for High-Resolution Remote Sensing Imagery Semantic Segmentation
    Chen, Jie
    Zhu, Jingru
    Sun, Geng
    Li, Jianhui
    Deng, Min
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (11) : 1921 - 1925
  • [40] Sea and Land Segmentation of Optical Remote Sensing Images Based on U-Net Optimization
    Li, Jianfeng
    Huang, Zhenghong
    Wang, Yongling
    Luo, Qinghua
    REMOTE SENSING, 2022, 14 (17)