AdaptiveWeighted Attention Network with Camera Spectral Sensitivity Prior for Spectral Reconstruction from RGB Images

Jiaojiao Li,Chaoxiong Wu,Rui Song,Yunsong Li,Fei Liu
DOI: https://doi.org/10.1109/cvprw50498.2020.00239
2020-01-01
Abstract:Recent promising effort for spectral reconstruction (SR) focuses on learninga complicated mapping through using a deeper and wider convolutional neuralnetworks (CNNs). Nevertheless, most CNN-based SR algorithms neglect to explorethe camera spectral sensitivity (CSS) prior and interdependencies amongintermediate features, thus limiting the representation ability of the networkand performance of SR. To conquer these issues, we propose a novel adaptiveweighted attention network (AWAN) for SR, whose backbone is stacked withmultiple dual residual attention blocks (DRAB) decorating with long and shortskip connections to form the dual residual learning. Concretely, we investigatean adaptive weighted channel attention (AWCA) module to reallocate channel-wisefeature responses via integrating correlations between channels. Furthermore, apatch-level second-order non-local (PSNL) module is developed to capturelong-range spatial contextual information by second-order non-local operationsfor more powerful feature representations. Based on the fact that the recoveredRGB images can be projected by the reconstructed hyperspectral image (HSI) andthe given CSS function, we incorporate the discrepancies of the RGB images andHSIs as a finer constraint for more accurate reconstruction. Experimentalresults demonstrate the effectiveness of our proposed AWAN network in terms ofquantitative comparison and perceptual quality over other state-of-the-art SRmethods. In the NTIRE 2020 Spectral Reconstruction Challenge, our entriesobtain the 1st ranking on the Clean track and the 3rd place on the Real Worldtrack. Codes are available at https://github.com/Deep-imagelab/AWAN.
What problem does this paper attempt to address?