Attention and transformer complementary fusion network for hyperspectral image spectral reconstruction

被引:1
作者
Zou, Changwu [1 ]
Zhang, Can [2 ]
Zou, Changzhong [2 ]
机构
[1] Fuzhou Univ, Coll Math & Stat, Fuzhou, Peoples R China
[2] Fuzhou Univ, Coll Comp & Data Sci, Fuzhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral imaging; image reconstruction; transformer; spatial attention; image resolution;
D O I
10.1080/01431161.2024.2371086
中图分类号
TP7 [遥感技术];
学科分类号
081102 ; 0816 ; 081602 ; 083002 ; 1404 ;
摘要
Hyperspectral images can be widely used in many fields due to their high information richness. However, costly and complex imaging spectrometers limit its growth. Hyperspectral reconstruction aims to obtain the corresponding hyperspectral image from the multispectral image, to reduce the acquisition cost of hyperspectral images. At present, the related works mainly use methods based on deep learning, and they have achieved good results. However, how to fully extract the global spectral and spatial features of hyperspectral images is still not well solved. To address these issues, we propose an Attention and Transformer Complementary Fusion Network (ATCFNet), which is composed of three modules: Multi-angle Input Image Processing (MIIP), Deep Feature Extraction (DFE) and Hyperspectral Reconstruction (HR) modules. Within the DFE module, an improved Transformer module and a novel Multi-scale Spatial Attention (MSA) module are proposed to extract the global spectral relationship and the spatial features of the hyperspectral images, respectively. Moreover, the MIIP module is proposed to extract the features of input multispectral images more effectively and comprehensively. To verify these, we compare the proposed ATCFNet with other excellent reconstruction methods on three hyperspectral image datasets. The results show that our method achieves the best results in these datasets. Code is publicly available at https://github.com/kidder314/ATCFNet.
引用
收藏
页码:5095 / 5112
页数:18
相关论文
共 34 条
[1]   In Defense of Shallow Learned Spectral Reconstruction from RGB Images [J].
Aeschbacher, Jonas ;
Wu, Jiqing ;
Timofte, Radu .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW 2017), 2017, :471-479
[2]   Sparse Recovery of Hyperspectral Signal from Natural RGB Images [J].
Arad, Boaz ;
Ben-Shahar, Ohad .
COMPUTER VISION - ECCV 2016, PT VII, 2016, 9911 :19-34
[3]   Using Machine Learning Algorithms With In Situ Hyperspectral Reflectance Data to Assess Comprehensive Water Quality of Urban Rivers [J].
Cai, Jiannan ;
Chen, Jun ;
Dou, Xianhui ;
Xing, Qianguo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[4]  
Can Y. B., 2018, PREPRINT, DOI [10.48550/arXivPreprintarXiv:18040.4647, DOI 10.48550/ARXIVPREPRINTARXIV:18040.4647]
[5]   HYPERSPECTRAL CLASSIFICATION USING LOW RANK AND SPARSITY MATRICES DECOMPOSITION [J].
Cao, Hongju ;
Shang, Xiaodi ;
Yu, Chunyan ;
Song, Meiping ;
Chang, Chein-, I .
IGARSS 2020 - 2020 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2020, :477-480
[6]  
Foucher PY, 2016, INT GEOSCI REMOTE SE, P7694, DOI 10.1109/IGARSS.2016.7731006
[7]  
Freeman J, 1997, P ANN INT IEEE EMBS, V19, P700, DOI 10.1109/IEMBS.1997.757727
[8]   CA-Net: Comprehensive Attention Convolutional Neural Networks for Explainable Medical Image Segmentation [J].
Gu, Ran ;
Wang, Guotai ;
Song, Tao ;
Huang, Rui ;
Aertsen, Michael ;
Deprest, Jan ;
Ourselin, Sebastien ;
Vercauteren, Tom ;
Zhang, Shaoting .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (02) :699-711
[9]   DsTer: A dense spectral transformer for remote sensing spectral super-resolution [J].
He, Jiang ;
Yuan, Qiangqiang ;
Li, Jie ;
Xiao, Yi ;
Liu, Xinxin ;
Zou, Yun .
INTERNATIONAL JOURNAL OF APPLIED EARTH OBSERVATION AND GEOINFORMATION, 2022, 109
[10]   A Very Deep Spatial Transformer Towards Robust Single Image Super-Resolution [J].
Jiang, Jianmin ;
Kasem, Hossam M. ;
Hung, Kwok-Wai .
IEEE ACCESS, 2019, 7 :45618-45631