[发明专利]基于自注意力机制的光谱超分辨率重建方法及系统

权利要求书

1.一种基于自注意力机制的光谱超分辨率重建方法，其特征在于，包括：

S101，对输入的RGB图像X进行特征提取得到空谱浅层特征H₀；

S102，将空谱浅层特征H₀输入训练好的多阶段光谱超分辨率重建网络MSCT得到重建高光谱图像H；所述多阶段光谱超分辨率重建网络MSCT由多个级联连接的单阶段光谱超分辨率重建网络SCT将空谱浅层特征H₀进行特征提取后，再通过一个卷积操作后与空谱浅层特征H₀叠加得到重建高光谱图像H，所述单阶段光谱超分辨率重建网络SCT用于通过一包含多级编码器的编码器模块对空谱浅层特征H₀进行编码、再通过一瓶颈模块将编码特征H_b送入一包含多级解码器的解码器模块，同时通过一基于自注意力机制的通道特征交叉融合模块CCT将各级编码器的编码特征H_i进行通道特征交叉融合得到交叉融合编码特征H_i^c并送入对应层级的解码器进行融合以最终获得输出的重建高光谱图像H。

2.根据权利要求1所述的基于自注意力机制的光谱超分辨率重建方法，其特征在于，所述通道特征交叉融合模块CCT将各级编码器的编码特征H_i进行通道特征交叉融合得到编码特征H_i^c包括：

S201，分别将各级编码器的编码特征H_i通过层归一化得到特征图T_i；

S202，将各级特征图T_i进行通道维堆叠操作得到特征图T；

S203，将各级特征图T_i以及特征图T通过多头通道自注意力机制单元MCA进行交叉通道自注意力计算得到各级注意力特征图T_i,out；

S204，分别将各级注意力特征图T_i,out首先与编码特征H_i跳跃连接，然后再将跳跃连接得到的特征依次经过层归一化、多层感知机后得到的特征图与原始的注意力特征图T_i,out进行跳跃连接后得到交叉融合编码特征H_i^c。