一种宽谱图像色彩重建方法

本发明涉及一种宽谱图像色彩重建方法。该方法具体步骤为：构建宽谱图像色彩重建网络，包括主干网络和语义子网络；获取宽谱语义分割数据集，预训练语义子网络；宽谱图像分两路分别输入主干网络和语义子网络，将语义子网络的中间输出通过语义引导交叉注意力模块连接到主干网络中，与主干网络的中间特征做交叉注意力操作；固定语义子网络的权重，仅更新主干网络的参数，在宽谱图像重建数据集上训练主干网络直至网络收敛；放开语义子网络的权重，使得语义子网络的参数和主干网络的参数在训练中一同更新，联合训练语义子网络和主干网络直至收敛；利用训练好的语义子网络和主干网络对宽谱图像进行重建恢复。本发明能有效解决宽谱成像中的色彩偏移问题。