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 条
  • [41] Segmentation of Mammogram Images Using U-Net with Fusion of Channel and Spatial Attention Modules (U-Net CASAM)
    Robert Singh, A.
    Vidya, S.
    Hariharasitaraman, S.
    Athisayamani, Suganya
    Hsu, Fang Rong
    Lecture Notes in Networks and Systems, 2024, 966 LNNS : 435 - 448
  • [42] Segmentation of Remote Sensing Images Based on U-Net Multi-Task Learning
    Ruiwen, Ni
    Ye, Mu
    Ji, Li
    Tong, Zhang
    Tianye, Luo
    Ruilong, Feng
    He, Gong
    Tianli, Hu
    Yu, Sun
    Ying, Guo
    Shijun, Li
    Tyasi, Thobela Louis
    Computers, Materials and Continua, 2022, 73 (02): : 3263 - 3274
  • [43] Individual tree crown delineation in high-resolution remote sensing images based on U-Net
    Maximilian Freudenberg
    Paul Magdon
    Nils Nölke
    Neural Computing and Applications, 2022, 34 : 22197 - 22207
  • [44] Individual tree crown delineation in high-resolution remote sensing images based on U-Net
    Freudenberg, Maximilian
    Magdon, Paul
    Noelke, Nils
    NEURAL COMPUTING & APPLICATIONS, 2022, 34 (24): : 22197 - 22207
  • [45] Segmentation of Remote Sensing Images Based on U-Net Multi-Task Learning
    Ni Ruiwen
    Mu Ye
    Li Ji
    Zhang Tong
    Luo Tianye
    Feng Ruilong
    Gong He
    Hu Tianli
    Sun Yu
    Guo Ying
    Li Shijun
    Tyasi, Thobela Louis
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 73 (02): : 3263 - 3274
  • [46] A remote sensing image classification procedure based on multilevel attention fusion U-Net
    Li D.
    Guo H.
    Lu J.
    Zhao C.
    Lin Y.
    Yu D.
    Cehui Xuebao/Acta Geodaetica et Cartographica Sinica, 2020, 49 (08): : 1051 - 1064
  • [47] Improved U-Net Network Segmentation Method for Remote Sensing Image
    Zhong, Letian
    Lin, Yong
    Sul, Yian
    Fang, Xianbao
    2022 IEEE 6TH ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2022, : 1034 - 1039
  • [48] DIResUNet: Architecture for multiclass semantic segmentation of high resolution remote sensing imagery data
    Sravya Priyanka
    Shyam N
    J Lal
    Chintala Sudhakar Nalini
    Fabio Reddy
    Applied Intelligence, 2022, 52 : 15462 - 15482
  • [49] DIResUNet: Architecture for multiclass semantic segmentation of high resolution remote sensing imagery data
    Priyanka
    Sravya, N.
    Lal, Shyam
    Nalini, J.
    Reddy, Chintala Sudhakar
    Dell'Acqua, Fabio
    APPLIED INTELLIGENCE, 2022, 52 (13) : 15462 - 15482
  • [50] SCAttNet: Semantic Segmentation Network With Spatial and Channel Attention Mechanism for High-Resolution Remote Sensing Images
    Li, Haifeng
    Qiu, Kaijian
    Chen, Li
    Mei, Xiaoming
    Hong, Liang
    Tao, Chao
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (05) : 905 - 909