基于深度学习的高光谱图像无损压缩方法技术

本发明专利技术公开了一种基于深度学习的高光谱图像无损压缩方法，解决了传统方法光谱信息利用不充分和模型泛化能力低的问题。实现步骤包括：应用深度学习中循环神经网络建立预测模型，对高光谱图像中的每个像素进行预测训练，生成预测图像和预测网络；对高光谱图像和预测图像做差，生成残差图像；对残差图像进行算术编码，生成码流文件；对码流文件进行解码得到解码图像；使用训练好的网络进行预测，得到预测图像；将预测图像与解码图像相加，得到原始的高光谱图像。本发明专利技术将深度学习与传统方法相结合，网络中采用有记忆性结构的单元，通过大量的训练，充分利用了光谱信息，增加了模型的泛化能力，提高了压缩效率。应用在高光谱图像压缩领域。

【技术实现步骤摘要】
基于深度学习的高光谱图像无损压缩方法
本专利技术属于图像处理
，涉及图像的无损压缩，具体是一种基于深度学习的高光谱图像的无损压缩方法，用于高光谱图像压缩。
技术介绍
伴随着高光谱遥感技术的快速发展，高光谱图像的空间分辨率与谱间分辨率也在不断提高，导致高光谱数据的规模持续增大。通常成像光谱仪均采用机载或者星载的方式获取影响数据，那么对于卫星数据链路而言，海量的高光谱影像数据获取与传输之间的矛盾日益加剧，从而制约高光谱遥感技术的发展。因此，为了高光谱遥感技术的长远发展，研究高效的高光谱压缩技术就显得尤为重要。目前，高光谱图像压缩方法主要是利用当前像素值的邻域像素值对其进行线性或非线性预测，通过对预测差值(预测值与真实值的差)的编码，使原本需要传输的码率降低，从而达到压缩的目的。其中，预测方法中使用比较广泛的是基于差分脉冲编码(DPCM)预测，该方法通过谱聚类预处理，然后在每个类中实现谱间最优线性预测，最后对预测残差进行熵编码，获得了当时几乎最高的无损压缩效果。DPCM实现了高光谱图像的无损压缩，但是其方法对用于预测的光谱波段的信息利用不是很充分，模型的泛化能力以及预测准确率都有待提高。
技术实现思路
本专利技术的目的在于克服上述已有技术的缺点和不足，提出一种充分利用光谱波段信息，泛化能力强的基于深度学习的高光谱图像无损压缩方法。本专利技术是一种基于深度学习的高光谱图像无损压缩方法，其特征在于，包括有如下步骤：(1)对原始的高光谱图像中的所有谱线进行聚类：输入待压缩的高光谱图像，该图像有M个波段；(2)对聚类后的谱线应用深度循环神经网络进行预测：整体采用分段预测，第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，第二波段到第M波段分别将深度循环神经网络和传统的最小二乘法作为预测器进行预测，得到初始预测图像和过渡预测图像；(3)应用预测出的图像和原始图像得出残差图像：将初始预测图像、过渡预测图像分别与原始高光谱图像相减得到初始残差图像和过渡残差图像；对比初始残差图像和过渡残差图像的每个像素，选择残差像素值较小的点，得出最终的残差图像；(4)对生成的残差图像进行算术编码，得到需要传输的码流文件；(5)对生成的码流文件进行解码得到还原的高光谱原始图像。本专利技术应用深度循环卷积网络与传统方法结合的方式保持信息的完整及正确性，提高压缩率，节约存储和传输空间。本专利技术与现有技术相比具有以下优点：a.本专利技术通过对高光谱图像建立预测模型进行压缩，以传统方式与深度学习结合的方式提高了压缩效率，同时通过利用当前波段的前N波段建立，建立预测模型，并调节预测模型中的阶数值，提高了压缩精度。b.本专利技术采用了深度循环神经网络作为预测器，其具有记忆性的结构克服了传统方法提取信息的局限性，将高光谱图像中数据的信息充分利用，得到较为准确的结果。c.本专利技术使用了具有记忆功能的LSTM模块作为深度循环卷积网络预测器的基本单元，其复杂的结构提高了模型提取特征的能力，从而充分利用了光谱信息。d.实验结果表明，本专利技术与C-DPCM等压缩方法相比，无损压缩性能较优。在本专利技术对五张图的平均压缩比特率比传统方法提高了0.09bpppb(bitsperpixelperband)。e.本专利技术通过聚类这种预处理方式，将光谱信息分类，简化了模型的训练过程，同时增加了模型的泛化能力。附图说明图1是本专利技术的流程图；图2是用本专利技术对一段光谱数据进行预测的结果；图3是本专利技术中使用的深度循环神经网络的结构图。具体实施方式下面结合附图和实例对本专利技术详细说明：实施例1目前，高光谱遥感技术正处于遥感技术的前沿领域，国内外均投入了大量的人力和物力资源对其进行研究，但是在高光谱成像仪获取的数据的实时传输方面还面临着巨大的困难，针对这一问题本专利技术从图像无损压缩的角度展开研究，着眼更高效的无损压缩方法，提出基于深度学习的高光谱图像无损压缩方法，参见图1，包括有如下步骤：(1)对原始的高光谱图像中的所有谱线进行聚类：输入待压缩的高光谱图像，该图像有M个波段。高光谱图像数据波段标准差大，信息量丰富；而波段间相关系数小，波段信息冗余度小，本例中波段数M为200。(2)对聚类后的谱线应用深度循环神经网络进行预测：整体采用分段预测，第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，第二波段到第M波段分别将深度循环神经网络和传统的最小二乘法作为预测器进行预测，得到初始预测图像和过渡预测图像。本例中的深度循环神经网络使用采用长短时间记忆(LSTM)模块，得出的结果中初始预测图像与过渡预测图像的规格均与原始高光谱图像相同。(3)应用预测出的图像和原始图像得出残差图像：将初始预测图像、过渡预测图像分别与原始高光谱图像运算得到初始残差图像和过渡残差图像，本例中直接采用相减的方法得出残差图像；对比初始残差图像和过渡残差图像的每个像素，选择残差像素值较小的点，得出最终的残差图像，即生成了最终的残差图像。(4)对生成的残差图像进行熵编码，得到需要传输的码流文件。熵编码是常用的无损压缩编码方式，它是一种编码过程中按熵原理不丢失任何信息的编码，本例中采用算术编码，将第一波段的第一个像素点与最终的残差图像编码为需要传输的码流文件。(5)对生成的码流文件进行解码得到还原的高光谱原始图像，解码方式是上述压缩编码的逆过程。本专利技术通过对高光谱图像建立预测模型进行压缩，以传统方式与深度学习结合的方式提高了压缩效率，同时通过利用当前波段的前N波段建立预测模型，并调节预测模型中的阶数值，提高了压缩精度。实施例2基于深度学习的高光谱图像无损压缩方法同实施例1，步骤(2)中所述的对聚类后的谱线应用深度循环神经网络进行预测，包括有如下步骤：(2a)对第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，即第一条谱线的第一波段像素值作为残差直接编码传输，第二条谱线的第一波段由第一条谱线预测，用这种方式得到所有谱线的第一波段。本例中所使用的相邻像素预测，即采用JPEG-LS压缩标准的核心算法LOCO-I进行波段内预测，其公式为(2b)从第二波段开始根据设定的预测阶数采用循环神经网络进行预测训练，当当前预测波段小于设定的预测阶数N时，采用当前波段前面的所有波段作为序列输入进行预测；当当前波段大于等于所设定的预测阶数N时，采用当前波段的前N个波段进行预测。预测阶数是压缩过程中重要的参数，当预测阶数较大时，压缩所需要的边信息就越大，压缩效果就越差；当预测阶数较小时，预测器算法能计算的信息越少，压缩效果也将下降，本例中预测阶数N为8。(2c)利用步骤(2b)中训练得到的网络，对所有的数据进行预测，从而得到原始图像对应的初始预测图像，其谱线参见图2，图2中Truedata为原始数据，prediction为本专利技术预测结果，图中可见本专利技术的预测结果与原始数据大致吻合。本例中训练样本数据中的高光谱波段数据作为训练输入，验证样本数据在训练网络的过程中每次迭代后进行测试，最终比较试验误差testloss和训练误差trainloss。当testloss不再降低时，终止网络训练。(2d)采用传统方法得到原始图像对应的过渡预测图像，本例中传统方法采用线性预测编码方法。步骤(2b)中使用的循环神经网络结构训练10epochs，使用动态学习率Ad本文档来自技高网...

【技术保护点】
1.一种基于深度学习的高光谱图像无损压缩方法，其特征在于，包括有如下步骤：(1)对原始的高光谱图像中的所有谱线进行聚类：输入待压缩的高光谱图像，该图像有M个波段；(2)对聚类后的谱线应用深度循环神经网络进行预测：整体采用分段预测，第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，第二波段到第M波段分别将深度循环神经网络和传统的最小二乘法作为预测器进行预测，得到初始预测图像和过渡预测图像；(3)应用预测出的图像和原始图像得出残差图像：将初始预测图像、过渡预测图像分别与原始高光谱图像相减得到初始残差图像和过渡残差图像；对比初始残差图像和过渡残差图像的每个像素，选择残差像素值较小的点，得出最终的残差图像；(4)对生成的残差图像进行算术编码，得到需要传输的码流文件；(5)对生成的码流文件进行解码得到还原的高光谱原始图像。

【技术特征摘要】
1.一种基于深度学习的高光谱图像无损压缩方法，其特征在于，包括有如下步骤：(1)对原始的高光谱图像中的所有谱线进行聚类：输入待压缩的高光谱图像，该图像有M个波段；(2)对聚类后的谱线应用深度循环神经网络进行预测：整体采用分段预测，第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，第二波段到第M波段分别将深度循环神经网络和传统的最小二乘法作为预测器进行预测，得到初始预测图像和过渡预测图像；(3)应用预测出的图像和原始图像得出残差图像：将初始预测图像、过渡预测图像分别与原始高光谱图像相减得到初始残差图像和过渡残差图像；对比初始残差图像和过渡残差图像的每个像素，选择残差像素值较小的点，得出最终的残差图像；(4)对生成的残差图像进行算术编码，得到需要传输的码流文件；(5)对生成的码流文件进行解码得到还原的高光谱原始图像。2.根据权利要求1所述的基于深度学习的高光谱图像无损压缩方法，其特征在于，其中步骤(2)中所述的对聚类后的谱线应用深度循环神经网络进行预测，包括有如下步骤：(2a)对第一波段采用谱内预测即采用相邻像素预测的方式进行波段内预测，即第一条谱线的第一波段像素值作为残差直接编码传输，第二条谱线的第一波段由第一条谱线预测，用这种方式得到所有谱线的第一波段；(2b)从第二波段开始根据设定的预测阶数采用循环神经网络进行预测训练，当当前预测波段小于设定的预测阶数N时，采用当前波段前面的所有波段作为序列输入进行预测；当当前波段大于等于所设定的预测阶数N时，采用当前波段的前N个波段进行预测；(2c)利用步骤(2b)中训练得到...

【专利技术属性】
技术研发人员：吴家骥，赵世慧，屈檀，李皎皎，
申请(专利权)人：西安电子科技大学，
类型：发明
国别省市：陕西,61