12 September 2022 Attention U-shaped network for hyperspectral image classification
Ruirui Wang, Bing Liu, Anzhu Yu, Wenjie Wang, Xuejun Jiao
Author Affiliations +
Abstract

To fully use the contextual information of hyperspectral images (HSIs), we propose a U-shaped network model combined with attention mechanism to achieve image-level HSI classification. First, the entire HSI is input into the network for end-to-end training, and the classification results of the entire scene are directly output. Then, the context information is used to improve the classification accuracy, while reducing many redundant calculations. Second, to improve the classification accuracy, considering two dimensions (i.e., space and channel), a hybrid attention module, mixing spatial and channel, is designed. Third, three datasets of the University of Pavia, Indian Pines, and Salinas are selected for the classification experiments. The experimental results show that, compared with other methods, the proposed method can obtain higher classification accuracy, and its training and testing efficiency is higher.

© 2022 Society of Photo-Optical Instrumentation Engineers (SPIE)
Ruirui Wang, Bing Liu, Anzhu Yu, Wenjie Wang, and Xuejun Jiao "Attention U-shaped network for hyperspectral image classification," Journal of Applied Remote Sensing 16(3), 036515 (12 September 2022). https://doi.org/10.1117/1.JRS.16.036515
Received: 16 March 2022; Accepted: 25 August 2022; Published: 12 September 2022
Lens.org Logo
CITATIONS
Cited by 2 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image classification

Hyperspectral imaging

Computer programming

Convolution

Remote sensing

Scene classification

Classification systems

Back to Top