Spectral-Spatial Attention Rotation-Invariant Classification Network for Airborne Hyperspectral Images | |
Shi, Yuetian1,2; Fu, Bin1,2![]() ![]() ![]() | |
作者部门 | 光谱成像技术研究室 |
2023-04 | |
发表期刊 | DRONES
![]() |
ISSN | 2504-446X |
卷号 | 7期号:4 |
产权排序 | 1 |
摘要 | An airborne hyperspectral imaging system is typically equipped on an aircraft or unmanned aerial vehicle (UAV) to capture ground scenes from an overlooking perspective. Due to the rotation of the aircraft or UAV, the same region of land cover may be imaged from different viewing angles. While humans can accurately recognize the same objects from different viewing angles, classification methods based on spectral-spatial features for airborne hyperspectral images exhibit significant errors. The existing methods primarily involve incorporating image or feature rotation angles into the network to improve its accuracy in classifying rotated images. However, these methods introduce additional parameters that need to be manually determined, which may not be optimal for all applications. This paper presents a spectral-spatial attention rotation-invariant classification network for the airborne hyperspectral image to address this issue. The proposed method does not require the introduction of additional rotation angle parameters. There are three modules in the proposed framework: the band selection module, the local spatial feature enhancement module, and the lightweight feature enhancement module. The band selection module suppresses redundant spectral channels, while the local spatial feature enhancement module generates a multi-angle parallel feature encoding network to improve the discrimination of the center pixel. The multi-angle parallel feature encoding network also learns the position relationship between each pixel, thus maintaining rotation invariance. The lightweight feature enhancement module is the last layer of the framework, which enhances important features and suppresses insignificance features. At the same time, a dynamically weighted cross-entropy loss is utilized as the loss function. This loss function adjusts the model's sensitivity for samples with different categories according to the output in the training epoch. The proposed method is evaluated on five airborne hyperspectral image datasets covering urban and agricultural regions. Compared with other state-of-the-art classification algorithms, the method achieves the best classification accuracy and is capable of effectively extracting rotation-invariant features for urban and rural areas. |
关键词 | airborne hyperspectral image hyperspectral image classification rotation-invariant local spatial feature enhancement convolutional neural network attention mechanism lightweight feature enhancement |
DOI | 10.3390/drones7040240 |
收录类别 | SCI |
语种 | 英语 |
WOS记录号 | WOS:000977540700001 |
出版者 | MDPI |
引用统计 | |
文献类型 | 期刊论文 |
条目标识符 | http://ir.opt.ac.cn/handle/181661/96462 |
专题 | 光谱成像技术研究室 |
通讯作者 | Zhang, Geng |
作者单位 | 1.Chinese Acad Sci, Xian Inst Opt & Precis Mech, Key Lab Spectral Imaging Technol CAS, Xian 710100, Peoples R China 2.Univ Chinese Acad Sci, Beijing 100049, Peoples R China 3.Xian Univ Posts & Telecommun, Sch Telecommun & Informat Engn, Xian 710061, Peoples R China |
推荐引用方式 GB/T 7714 | Shi, Yuetian,Fu, Bin,Wang, Nan,et al. Spectral-Spatial Attention Rotation-Invariant Classification Network for Airborne Hyperspectral Images[J]. DRONES,2023,7(4). |
APA | Shi, Yuetian.,Fu, Bin.,Wang, Nan.,Cheng, Yinzhu.,Fang, Jie.,...&Zhang, Geng.(2023).Spectral-Spatial Attention Rotation-Invariant Classification Network for Airborne Hyperspectral Images.DRONES,7(4). |
MLA | Shi, Yuetian,et al."Spectral-Spatial Attention Rotation-Invariant Classification Network for Airborne Hyperspectral Images".DRONES 7.4(2023). |
条目包含的文件 | ||||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 | ||
Spectral-Spatial Att(20465KB) | 期刊论文 | 出版稿 | 限制开放 | CC BY-NC-SA | 请求全文 |
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。
修改评论