• Optoelectronics Letters
  • Vol. 20, Issue 10, 599 (2024)
Jiamin YU1, Sixian CHAN1,2,3,*, Yanjing LEI1, Wei WU1,3..., Yuan WANG1 and Xiaolong ZHOU4|Show fewer author(s)
Author Affiliations
  • 1College of Computer Science and Technology, Zhejiang University of Technology, Hangzhou 310023, China
  • 2Key Laboratory of Meteorological Disaster (KLME), Ministry of Education & Collaborative Innovation Center on Forecast and Evaluation of Meteorological Disasters (CIC-FEMD), Nanjing University of Information Science &Technology, Nanjing 210044, China
  • 3College of Geographic Information Modern Industry, Zhejiang University of Technology, Hangzhou 310023, China
  • 4College of Electrical and Information Engineering, Quzhou University, Quzhou 324000, China2
  • show less
    DOI: 10.1007/s11801-024-3179-1 Cite this Article
    YU Jiamin, CHAN Sixian, LEI Yanjing, WU Wei, WANG Yuan, ZHOU Xiaolong. 2SWUNet: small window SWinUNet based on tansformer for building extraction from high-resolution remote sensing images[J]. Optoelectronics Letters, 2024, 20(10): 599 Copy Citation Text show less

    Abstract

    Models dedicated to building long-range dependencies often exhibit degraded performance when transferred to remote sensing images. Vision transformer (ViT) is a new paradigm in computer vision that uses multi-head self-attention(MSA) rather than convolution as the main computational module, with global modeling capabilities. However, its performance on small datasets is usually far inferior to that of convolutional neural networks (CNNs). In this work, we propose a small window SWinUNet (2SWUNet) for building extraction from high-resolution remote sensing images.Firstly, the 2SWUNet is trained based on swin transformer by designing a fully symmetric encoder-decoder U-shaped architecture. Secondly, to construct a reasonable U-shaped architecture for building extraction from high-resolution remote sensing images, different forms of patch expansion are explored to simulate up-sampling operations and recover feature map resolution. Then, the small window-based multi-head self-attention (W-MSA) is designed to reduce the computational and memory burden, which is more appropriate for the features of remote sensing images. Meanwhile, the pre-training mechanism is advanced to make up for the lack of decoder parameters. Finally, comparison experiments with other mainstream CNNs and ViTs validate the superiority of the proposed model.
    YU Jiamin, CHAN Sixian, LEI Yanjing, WU Wei, WANG Yuan, ZHOU Xiaolong. 2SWUNet: small window SWinUNet based on tansformer for building extraction from high-resolution remote sensing images[J]. Optoelectronics Letters, 2024, 20(10): 599
    Download Citation