Akademska digitalna zbirka SLovenije - logo
E-viri
Celotno besedilo
Recenzirano
  • Rotation-Invariant Attentio...
    Zheng, Xiangtao; Sun, Hao; Lu, Xiaoqiang; Xie, Wei

    IEEE transactions on image processing, 01/2022, Letnik: 31
    Journal Article

    Hyperspectral image (HSI) classification refers to identifying land-cover categories of pixels based on spectral signatures and spatial information of HSIs. In recent deep learning-based methods, to explore the spatial information of HSIs, the HSI patch is usually cropped from original HSI as the input. And <inline-formula> <tex-math notation="LaTeX">3 \times 3 </tex-math></inline-formula> convolution is utilized as a key component to capture spatial features for HSI classification. However, the <inline-formula> <tex-math notation="LaTeX">3 \times 3 </tex-math></inline-formula> convolution is sensitive to the spatial rotation of inputs, which results in that recent methods perform worse in rotated HSIs. To alleviate this problem, a rotation-invariant attention network (RIAN) is proposed for HSI classification. First, a center spectral attention (CSpeA) module is designed to avoid the influence of other categories of pixels to suppress redundant spectral bands. Then, a rectified spatial attention (RSpaA) module is proposed to replace <inline-formula> <tex-math notation="LaTeX">3 \times 3 </tex-math></inline-formula> convolution for extracting rotation-invariant spectral-spatial features from HSI patches. The CSpeA module, the <inline-formula> <tex-math notation="LaTeX">1 \times 1 </tex-math></inline-formula> convolution and the RSpaA module are utilized to build the proposed RIAN for HSI classification. Experimental results demonstrate that RIAN is invariant to the spatial rotation of HSIs and has superior performance, e.g., achieving an overall accuracy of 86.53% (1.04% improvement) on the Houston database. The codes of this work are available at https://github.com/spectralpublic/RIAN .