基于三维协同注意力的高光谱图像压缩与重建方法

本发明基于三维协同注意力的高光谱图像压缩与重建方法，包括选取高光谱图像数据集进行预处理，将预处理后的数据集划分为训练集、验证集和测试集，建立由编码器和解码器构成的高光谱图像压缩网络模型，编码器包含三维协同注意力模块，解码器以特征增强主干架构为基础，配置有特征精炼模块，对高光谱图像压缩网络模型进行训练、验证和测试，获得优化后的高光谱图像压缩网络模型，将待压缩的图像输入优化后的高光谱图像压缩网络模型中，实现对高光谱图像的压缩与重建。本发明通过引入三维协同注意力机制在编码阶段实现空间–光谱–通道信息的融合表达，从整体上提升了压缩特征的语义完整性与判别能力，提高了模型的压缩性能。