[发明专利]高光谱图像压缩方法、装置、设备及可读存储介质

说明书

本发明提供一种高光谱图像压缩方法、装置、设备及可读存储介质。该方法包括：通过训练集对卷积神经网络进行训练，其中，卷积神经网络包括非线性变换模块、量化模块以及熵模型；使用测试集验证训练完成的卷积神经网络的压缩性能，当训练完成的卷积神经网络的压缩性能达标时，通过训练完成的卷积神经网络对高光谱图像进行压缩处理。通过本发明，对于高光谱图像压缩具有较好的率失真性能。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种高光谱图像压缩方法、装置、设备及可读存储介质。

背景技术

高光谱图像具有丰富而独特的光谱信息，给基于高光谱图像的很多应用，比如农作物分类、质量检测、灾难预测等任务带来了极大的便利。然而这种优势在有限的传输带宽和存储能力下，也制约着高光谱图像的进一步发展。因此，如何有效解决高光谱图像大数据量带来的各种挑战是高光谱图像得以广泛应用的前提和关键。

在高光谱图像压缩算法中，变换编码因其较小的计算复杂度和良好的适应性而得到广泛应用。基于变换编码的图像压缩算法包括四部分：变换，量化和熵编码，反变换，分开实现编码过程和去相关过程。

变换是通过一定的方式将图像从像素域变换到一个更为紧凑的空间，现有基于变换编码的高光谱图像压缩方法一般假设高光谱图像为高斯源，在此条件下只需可逆的线性变换即能将像素映射成独立的潜在表示，通过量化和熵编码将潜变量压缩成码流用于存储和传输。然而，实际场景的高光谱图像具有明显的非高斯特性，使得线性变换不再适用，非线性变换的探索为该问题提供了新的方法和思路。近年来，以人工神经网络尤其是深度学习为工具的非线性变换的发展改变了传统图像压缩手工设参的状况。现有基于深度学习的图像压缩技术表现出巨大的潜力，性能已超过工业界的H.266/VVC(Versatile VideoCoding，VVC)标准。然而，这些方法多用于处理三波段的自然图像，对于高光谱图像的压缩相对较少。

变换过程使得量化和和熵编码得以在一个紧凑空间执行，相较于RGB自然图像，高光谱图像的光谱间具有更强的相关性，这使得经过相同的变换过程，高光谱图像得到的潜在表示具有与RGB图像不同的统计特性。经过量化，潜变量变成离散形式，然后基于熵编码算法对该离散形式进行编码。而熵编码的过程依赖于潜变量的概率分布模型，由最小熵理论可知，设计的熵模型越接近真实的潜变量分布，码率越小，熵率优化过程中得到的解越接近最优解。

结合上述分析，目前基于深度学习的压缩技术还需要进一步根据高光谱图像的特点，设计更加灵活准确的熵模型，以减少熵模型与真实潜变量分布之间的失配，从而达到最优的率失真性能。

发明内容

为解决上述技术问题，本发明提供一种高光谱图像压缩方法、装置、设备及可读存储介质。

第一方面，本发明提供一种高光谱图像压缩方法，所述高光谱图像压缩方法包括：

通过训练集对卷积神经网络进行训练，其中，卷积神经网络包括非线性变换模块、量化模块以及熵模型；

使用测试集验证训练完成的卷积神经网络的压缩性能，当训练完成的卷积神经网络的压缩性能达标时，通过训练完成的卷积神经网络对高光谱图像进行压缩处理。

可选的，在所述通过训练集对卷积神经网络进行训练的步骤之前，还包括：

将样本高光谱图像在空间维切分成若干固定大小的立方块；

按预设比例将所述若干固定大小的立方块划分为训练集和测试集。

可选的，非线性变换模块对高光谱图像的空间和光谱维进行正向非线性变换，得到潜变量；量化模块通过添加均匀噪声的方式对潜变量进行量化；熵模型用于得到潜变量的概率分布，以供在熵编码时基于所述概率分布确定潜变量中每个元素所分配的码字。

可选的，基于率失真准则对卷积神经网络训练过程进行约束，用来确定非线性变换模块以及熵模型中的参数值。