火星图像压缩方法、装置、计算机设备和存储介质

1.本技术涉及图像处理技术领域，尤其涉及一种火星图像压缩方法、装置、计算机设备和存储介质。


背景技术：

2.图像压缩是消除冗余数据的过程，在通信带宽较小的情况下，需要使用高效的图像压缩方法对原始图像数据进行压缩，以减小表征图像所需要的数据量，提高传输效率。
3.目前，常见的图像压缩包括无损压缩和有损压缩，在实际应用中，常见的有损图像压缩方法是基于jpeg、jpeg2000、hevc等压缩标准的图像压缩方法，但这些图像压缩方法的编码工具比较复杂，很难实现所有工具的联合优化。随着深度学习的迅速发展，端到端图像压缩技术实现了整个编码框架的联合优化，但火星图像的非局部相似度较高，目前的图像压缩方法并不能实现对火星图像的高效压缩。因此如何实现对火星图像的高效压缩是亟待解决的问题。


技术实现要素：

4.为了解决现有的图像压缩技术不能对非局部相似度较高的火星图像进行高效压缩的技术问题，本技术提供一种火星图像压缩方法、装置、计算机设备和存储介质。
5.第一方面，本技术实施例提供一种火星图像压缩方法，所述方法包括：
6.获取待压缩火星图像；
7.根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩；其中，所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征；
8.若是，则将所述待压缩火星图像输入所述目标火星图像压缩模型，获取所述待压缩火星图像对应的压缩图像。
9.在本技术一些实施例中，所述非局部模块包括非局部块、多个残差块，以及一个或多个卷积层；其中，所述非局部块与多个残差块连接，残差块之后连接一个卷积层，所述非局部块包括多个下采样单元和多个最大池化层。
10.在本技术一些实施例中，在根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩之前，所述方法还包括：获取所述待压缩火星图像的非局部相似度；
11.所述获取所述待压缩火星图像的非局部相似度，包括：
12.将所述待压缩火星图像分割为多个第一图像块，所述多个第一图像块互不重叠；
13.确定所述多个第一图像块之间的结构相似性或均方根误差，并根据所述结构相似性或所述均方根误差确定所述待压缩火星图像的非局部相似度。
14.在本技术一些实施例中，在根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩之前，所述方法还包括：获取所述待压缩火星图像的非局部相似度；
15.所述获取所述待压缩火星图像的非局部相似度，包括：
16.对所述待压缩火星图像加入加性高斯噪声，得到加噪火星图像；
17.将所述加噪火星图像分割为多个第二图像块，并将所述第二图像块中相似的图像块匹配在一起进行滤波处理，得到去噪火星图像；
18.获取所述待压缩火星图像和所述去噪火星图像的峰值信噪比，根据所述峰值信噪比确定所述待压缩火星图像的非局部相似度。
19.在本技术一些实施例中，在将所述待压缩火星图像输入目标火星图像压缩模型进行压缩之前，所述方法还包括：获取所述目标火星图像压缩模型；
20.其中，所述获取所述目标火星图像压缩模型包括：
21.在超先验模型中插入一个或多个非局部模块，构建初始火星图像压缩模型；
22.根据目标样本数据集中图像的非局部相似度与所述预设相似度阈值的关系获取训练样本集，并基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型。
23.在本技术一些实施例中，所述基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型，包括：
24.将所述训练图像输入所述初始火星图像压缩模型，获取所述训练图像对应的压缩图像；
25.根据所述训练图像对应的压缩图像的失真和编码码率对初始火星图像压缩模型的权重参数进行调整，直至预设损失函数收敛，得到所述目标火星图像压缩模型。
26.在本技术一些实施例中，所述方法还包括：获取验证数据集，并基于所述验证数据集对所述目标火星图像压缩模型的超参数进行调整，以提高所述目标火星图像压缩模型的准确率。
27.第二方面，本技术实施例提供一种火星图像压缩装置，包括：
28.采集模块，用于获取待压缩火星图像；
29.判断模块，用于根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩；其中，所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征；
30.处理模块，用于将所述待压缩火星图像输入所述目标火星图像压缩模型，获取所述待压缩火星图像对应的压缩图像。
31.在本技术一些实施例中，所述非局部模块包括非局部块、多个残差块，以及一个或多个卷积层；其中，所述非局部块与多个残差块连接，残差块之后连接一个卷积层，所述非局部块包括多个下采样单元和多个最大池化层。
32.在本技术一些实施例中，所述装置还包括计算模块；在根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待
压缩火星图像进行压缩之前，用于将所述待压缩火星图像分割为多个第一图像块，所述多个第一图像块互不重叠；
33.确定所述多个第一图像块之间的结构相似性或均方根误差，并根据所述结构相似性或所述均方根误差确定所述待压缩火星图像的非局部相似度。
34.在本技术一些实施例中，所述计算模块还用于对所述待压缩火星图像加入加性高斯噪声，得到加噪火星图像；
35.将所述加噪火星图像分割为多个第二图像块，并将所述第二图像块中相似的图像块匹配在一起进行滤波处理，得到去噪火星图像；
36.获取所述待压缩火星图像和所述去噪火星图像的峰值信噪比，根据所述峰值信噪比确定所述待压缩火星图像的非局部相似度。
37.在本技术一些实施例中，所述装置还包括构建模块，用于在将所述待压缩火星图像输入目标火星图像压缩模型进行压缩之前，获取所述目标火星图像压缩模型；
38.其中，所述获取所述目标火星图像压缩模型包括：
39.在超先验模型中插入一个或多个非局部模块，构建初始火星图像压缩模型；
40.根据目标样本数据集中图像的非局部相似度与所述预设相似度阈值的关系获取训练样本集，并基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型。
41.在本技术一些实施例中，所述构建模块，具体用于将所述训练图像输入所述初始火星图像压缩模型，获取所述训练图像对应的压缩图像；
42.根据所述训练图像对应的压缩图像的失真和编码码率对初始火星图像压缩模型的权重参数进行调整，直至预设损失函数收敛，得到所述目标火星图像压缩模型。
43.在本技术一些实施例中，所述装置还包括验证模块，用于获取验证数据集，并基于所述验证数据集对所述目标火星图像压缩模型的超参数进行调整，以提高所述目标火星图像压缩模型的准确率。
44.第三方面，本技术实施例提供了一种计算机设备，包括：存储器和处理器，存储器用于存储计算机程序；处理器用于在调用计算机程序时执行第一方面或第一方面任一种可选的实施方式所述的火星图像压缩方法的步骤。
45.第四方面，本技术实施例提供了一种计算机可读存储介质，其上存储有计算机程序，计算机程序被处理器执行时实现第一方面或第一方面任一种可选的实施方式所述的火星图像压缩方法的步骤。
46.本技术第一方面实施例提供的火星图像压缩方法包括：获取待压缩火星图像；根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩；若是，则将所述待压缩火星图像输入所述目标火星图像压缩模型，获取所述待压缩火星图像对应的压缩图像；所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征。本技术实施例中通过在目标火星图像压缩模型的编码器和解码器插入非局部模块，可以实现对待压缩火星图像的非局部特征进行提取，从而实现对非局部相似度较高的火星图像进行高效压缩，在编码码
率较小的情况下，保证火星图像压缩后的失真较小。
附图说明
47.此处的附图被并入说明书中并构成本说明书的一部分，示出了符合本技术的实施例，并与说明书一起用于解释本技术的原理。
48.为了更清楚地说明本技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，对于本领域普通技术人员而言，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
49.图1为本技术一个实施例提供的火星图像压缩方法的步骤流程图；
50.图2a为本技术一个实施例提供的目标火星图像压缩模型的结构框图；
51.图2b为本技术一个实施例提供的目标火星图像压缩模型中非局部模块的结构框图；
52.图3为本技术另一个实施例提供的火星图像压缩方法的步骤流程图；
53.图4为本技术一个实施例中非局部块的结构示意图；
54.图5a为本技术一个实施例中目标火星图像压缩模型对待压缩火星图像进行压缩后的结果对比示意图；
55.图5b为本技术另一个实施例中目标火星图像压缩模型对待压缩火星图像进行压缩后的结果对比示意图；
56.图6为本技术一个实施例中火星图像压缩装置的结构框图；
57.图7为本技术另一个实施例中火星图像压缩装置的结构框图；
58.图8为本技术一个实施例中计算机设备的内部结构图。
具体实施方式
59.为了使本技术的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本技术进行进一步详细说明。应当理解，此处描述的具体实施例仅仅用以解释本技术，并不用于限定本技术。需要说明的是，在不冲突的情况下，本技术的实施例及实施例中的特征可以相互组合。
60.在下面的描述中阐述了很多具体细节以便于充分理解本技术，但本技术还可以采用其他不同于在此描述的方式来实施；显然，说明书中的实施例只是本技术的一部分实施例，而不是全部的实施例。
61.本技术的说明书和权利要求书中的术语“第一”和“第二”等是用于区别同步的对象，而不是用于描述对象的特定顺序。例如，第一功能和第二功能是用于区别不同的功能，而不是用于描述功能的特定顺序。
62.在本技术实施例中，“示例性的”或者“例如”等词用于表示作例子、例证或说明。本技术实施例中被描述为“示例性的”或者“例如”的任何实施例或设计方案不应被解释为比其它实施例或设计方案更优选或更具优势。确切而言，使用“示例性的”或者“例如”等词旨在以具体方式呈现相关概念，此外，在本技术实施例的描述中，除非另有说明，“多个”的含义是指两个或两个以上。
63.首先，对本技术实施例的应用场景进行示例性描述：本技术实施例提供的火星图
像压缩方法可用于对非局部相似度较大的图像，本技术实施例以火星图像为例对本技术实施例提供的火星图像压缩方法进行说明，但需要说明的是，本技术实施例提供的火星图像压缩方法并不限于火星图像，也适用于其它非局部相似度较大的其它图像。
64.为支撑火星科学探索和研究，需将火星上获取的大量图像数据传输到地球。然而，由于地火通信距离远导致的通信带宽极窄，因此对这些大量数据进行压缩是必不可少的一步。目前，常见的针对图像的压缩标准有jpeg、jpeg2000、hevc等，但这些现有压缩方法编码性能的提升主要通过加入更复杂的编码工具才能实现，这些编码工具的复杂性使得很难实现所有编码工具的联合优化。此外，针对极低码率的传输场景，可利用生成对抗网络只传输图像的部分信息，直接合成消耗码率较多的细节信息，从而节省编码比特并实现了主观质量较好的压缩图像，但这种压缩方法并不能对非局部相似度较大的火星图像实现高效压缩。
65.火星图像压缩时编码性能的衡量包括编码码率和失真，本技术实施例旨在找到编码码率和失真之间的平衡，使对火星图像压缩的性能最为高效。
66.基于此，本技术实施例提供了一种火星图像压缩方法，目标火星图像压缩模型的第一编码器、第一解码器、第二编码器、以及第二解码器中均插入了非局部模块，用于提取待压缩火星图像的非局部特征，从而实现对非局部相似度较大的火星图像的高效压缩。
67.其中，本技术实施例提供的火星图像压缩方法可以通过本技术实施例提供的火星图像压缩装置或电子设备来执行，该电子设备可以为手机、平板电脑、笔记本电脑、超级移动个人计算机(ultra-mobile personal computer，umpc)等终端设备，或者还可以为其他类型的终端设备，本技术实施例不进行具体限定。
68.下面通过几个具体实施例，对本技术提供的火星图像压缩方法进行详细阐述。
69.图1为本技术一个实施例提供的火星图像压缩方法的流程图，参照图1所示，本实施例提供的火星图像压缩方法包括以下步骤：
70.s110、获取待压缩火星图像。
71.其中，本实施例不对待压缩火星图像的获取方式进行具体限定。
72.s120、根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩。
73.其中，所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征。
74.若是，则执行如下步骤s130。其中，预设相似度阈值可根据具体的应用场景确定。
75.需要说明的是，在本技术实施例中，待压缩火星图像的非局部相似度与预设相似度阈值的关系可理解为用于表示待压缩火星图像的非局部相似度的参数与该参数对应的预设参数阈值之间的关系，用于表示待压缩火星图像的非局部相似度的参数可以为一个或多个，如图像的结构相似性(structural similarity,ssim)、均方根误差(root mean square error,rmse)、峰值信噪比(peak signal to noise ratio,psnr)等等中的一个或多个。例如，待压缩火星图像的非局部相似度与预设相似度阈值的关系可以为待压缩火星图像的均方根误差与预设均方根误差阈值之间的关系，也可以为待压缩火星图像对应的峰
值信噪比与预设峰值信噪比阈值之间的关系。
76.可选的，在根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩之前，可首先获取所述待压缩火星图像的非局部相似度与预设相似度阈值之间的关系，若待压缩火星图像的非局部相似度与预设相似度阈值的关系满足预设条件，则确定利用目标火星图像压缩模型对所述待压缩火星图像进行压缩。所述预设条件为用于选择非局部相似度较高的火星图像的条件。
77.其中，获取所述待压缩火星图像的非局部相似度可通过以下几种方式实现。
78.第一种实现方式、将所述待压缩火星图像分割为多个第一图像块，示例性的，可切割为128
×
160的图像块。所述多个第一图像块互不重叠；确定所述多个第一图像块之间的结构相似性或均方根误差，并根据所述结构相似性或所述均方根误差确定所述待压缩火星图像的非局部相似度与预设相似度阈值的关系。
79.其中，结构相似性可用于表征第一图像块的轮廓、细节等的相似性。结构相似性越强、均方根误差越小，说明第一图像块之间的相似度越高，待压缩火星图像的非局部相似度越高。
80.示例性的，当以rmse表示待压缩火星图像的非局部相似度时，预设相似度阈值(预设rmse阈值)可以为69.3080，即当待压缩火星图像对应的rmse小于69.3080时，确定待压缩火星图像的非局部相似度与预设相似度阈值的关系满足预设条件，确定利用所述目标火星图像压缩模型对所述待压缩火星图像进行压缩。
81.第二种实现方式、对所述待压缩火星图像加入加性高斯噪声，得到加噪火星图像；将所述加噪火星图像分割为多个第二图像块，并将所述第二图像块中相似的图像块匹配在一起进行滤波处理，得到去噪火星图像；获取所述待压缩火星图像和所述去噪火星图像的峰值信噪比(peak signal to noise ratio,psnr)，根据所述峰值信噪比确定所述待压缩火星图像的非局部相似度与预设相似度阈值的关系。
82.示例性的，当以psnr表示待压缩火星图像的非局部相似度时，若加入加性高斯噪声的噪声强度的标准差为170时，预设相似度阈值可以为18.3443(预设psnr值)，即当待压缩火星图像对应的psnr大于18.3443时，确定待压缩火星图像的非局部相似度与预设相似度阈值的关系满足预设条件，确定利用所述目标火星图像压缩模型对所述待压缩火星图像进行压缩。待压缩火星图像对应的psnr越大，其非局部相似度越高。
83.此外，待压缩火星图像的非局部相似度还可基于rmse和psnr共同确定，即当rmse小于预设rmse阈值，且psnr大于预设psnr阈值时，待压缩火星图像的非局部相似度与预设相似度阈值的关系满足预设条件。
84.需要说明的是，所述预设相似度阈值可根据需要确定，以上只是作为示例性说明，不构成任何限定。
85.其中，将所述多个第二图像块中相似的图像块匹配在一起匹配可以理解为将相似的图像块进行重叠。
86.此外，可基于块匹配和三维滤波(block-matching and 3-dimesioanl filtering，bm3d)确定多个第二图像块中相似的图像块，示例性的，对于每个第二图像块，在附近寻找最多maxn1(超参数)个相似的第二图像块图块，为了避免噪点的影响，将图块经
过2d变换后再用欧氏距离衡量相似程度。
87.可选的，可通过bm3d去噪算法去除待压缩火星图像上的高斯噪声，去噪的性能由峰值信噪比进行评估，峰值信噪比越高，说明去噪效果越好，匹配在一起进行处理的第二图像块的相似度越高。
88.s130、将所述待压缩火星图像输入目标火星图像压缩模型进行压缩，获取所述待压缩火星图像对应的压缩图像。
89.其中，所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征。
90.示例性的，目标火星图像压缩模型的结构框图参照图2a所示，本技术实施例提供的火星图像压缩方法可基于图2a所示结构的模型对待压缩火星图像进行压缩。参照图2a所示，201为第一编码器，202为第一解码器，203为第二编码器(超先验编码器)，204为第二解码器(超先验解码器)，待压缩火星图像经所述第一编码器输入进行压缩，待压缩火星图像对应的压缩图像经第一解码器输出。其中，第一编码器和第一解码器用来压缩输入的待压缩火星图像，同时第二编码器和第二编解码器用于去除输入的待压缩火星图像的高维表征中存在的空域冗余，提升编码效率。其中，高维表征中的空域冗余即超先验，第二编码器和第二编解码器对所述待压缩火星图像进行压缩、重建，重建后的超先验可类比为传统压缩中的边信息，被输入第一编码器和第一解码器提升其熵编码效率。
91.参照图2a所示，conv表示卷积层，abs表示绝对值，采用广义归一化除法gdn作为激活函数，对每一位数据进行非线性映射；igdn为gdn的逆变换，与编码器中激活层相对应，q表示量化；ae表示算术编码，ad表示算术解码，二者都是熵编码的一部分；relu表示修正线性单元。其中，多个结合gdn和修正线性单元(rectified linear unit，relu)的卷积层将输入图像或超先验转换为潜在(高维)表征。由于量化和熵编码都是不可导的，因此在网络训练阶段，可导的加性均匀噪声用来近似量化，交叉熵近似熵编码。在网络测试阶段，采用真实的量化和熵编码获得最终的压缩图像和压缩码率。
92.非局部模块的结构框架参照图2b所示，sigmoid表示sigmoid激活函数。输入图像经过非局部块、以及非局部块之后的连接的多个残差块之后，通过卷积层执行卷积操作，得到卷积结果，将卷积结果通过激活函数映射后，与另一残差块的输出进行点乘操作，将点乘结果与非局部块的输入进行加和操作，得到输入图像对应的特征图。
93.本技术第一方面实施例提供的火星图像压缩方法包括：获取待压缩火星图像；根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩；若是，则将所述待压缩火星图像输入所述目标火星图像压缩模型，获取所述待压缩火星图像对应的压缩图像；所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征。本技术实施例中通过在目标火星图像压缩模型的编码器和解码器插入非局部模块，可以实现对待压缩火星图像的非局部特征进行提取，从而实现对非局部相似度较高的火星图像进行高效压缩，在编码码
率较小的情况下，保证了火星图像压缩后的失真较小。
94.参照图3所示，图3为本技术另一个实施例提供的火星图像压缩方法的流程图，结合图1所示，在图1所示实施例的基础上，步骤s110之前还包括构建目标火星图像压缩模型，示例性的，构建目标火星图像压缩模型可包括如下步骤：
95.s310、在超先验模型中插入一个或多个非局部模块，构建初始火星图像压缩模型。
96.其中，所述非局部模块包括非局部块、多个残差块，以及一个或多个卷积层；其中，所述非局部块与多个残差块连接，残差块之后连接一个卷积层，所述非局部块包括多个下采样单元和多个最大池化层。
97.示例性的，参照图4所示，图4为非局部块的结构示意图，该非局部块包括多个下采样单元和多个最大池化层，下采样单元用于将输入通道c下采样为c/s，最大池化层用于将分辨率h
×
w下采样为h
×
w/d2。
98.s320、根据目标样本数据集中图像的非局部相似度与所述预设相似度阈值的关系获取训练样本集，并基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型。
99.示例性的，当以rmse表示图像的非局部相似度时，将rmse小于预设相似度阈值(预设rmse阈值)的火星图像确定为训练图像加入训练样本集；当以psnr表示图像的非局部相似度时，将psnr大于预设相似度阈值(预设psnr阈值)的火星图像加入训练样本集。图像的rmse越小，psnr越大，该图像的非局部相似度越高。
100.在本技术一个实施例中，可通过获取火星图像构建训练样本集。或通过获取火星图像以及非局部相似度满足预设相似度阈值的自然图像构建训练样本集。
101.在训练样本集包括的训练图像为火星图像的情况下，获取训练样本集包括：将登陆火星的毅力号火星车采集的火星图像作为初始样本数据集，初始样本数据集包含大量的彩色高分辨率(分辨率为1152
×
1600)的无损彩色火星图像，在初始样本数据集中选取相机系统左目相机拍摄的火星图像、舍弃右目相机拍摄的火星图像，从而去除图像间的冗余。此外，还删除了包含火星车、火星车轮子轨迹、火星车钻孔的图像和模糊图像，以及对选择的火星图像进行居中切割去除图像四周的黑块，得到目标样本数据集。在所述目标样本数据集中确定各火星图像的非局部相似度，根据各火星图像的非局部相似度与预设相似度阈值的关系确定训练样本集、验证数据集和测试数据集，其中，训练样本集、验证数据集和测试数据集中均为非局部相似度较高的火星图像，其非局部相似度与预设相似度阈值的关系满足预设条件。此外，可按照8：1：1的比例在目标样本数据集中确定训练样本集、验证数据集和测试数据集。示例性的，在满足预设条件的火星图像的数量为3860张的情况下，训练样本集中的训练图像为3088张，验证数据集和测试数据集中各包括386张火星图像。
102.此外，为了扩大训练样本集的数据量，提高目标火星图像压缩模型的准确度，还可选取非局部相似度与预设相似度阈值的关系满足预设条件自然图像，加入训练样本集。
103.其中，预设条件即为用于表示图像的非局部相似度的参数与该参数对应的预设参数阈值之间应满足的关系。示例性的，在基于目标数据集确定训练样本集时，也可采用获取图像的结构相似性或均方根误差对图像的非局部相似度进行度量，当基于均方根误差进行度量时，若火星图像的均方根误差小于预设均方根误差阈值，则可将该火星图像确定为训练样本集中的训练图像。此外，还可基于均方根误差和峰值信噪比共同确定练样本集、验证
数据集和测试数据集，此处不再赘述。
104.或者，还可采用bm3d算法确定图像的非局部相似度，将非局部相似度满足预设条件的图像作为训练样本集中的训练图像。采用bm3d算法确定图像的非局部相似度的具体过程可参见前文描述，此处不再赘述。
105.示例性的，基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型可包括：
106.将所述训练图像输入所述初始火星图像压缩模型，获取所述训练图像对应的压缩图像；根据所述训练图像对应的压缩图像的失真和编码码率对初始火星图像压缩模型的权重参数进行调整，直至预设损失函数收敛，得到所述目标火星图像压缩模型。
107.其中，可将所述训练图像切割为512
×
512的图像块输入所述初始火星图像压缩模型，对所述初始火星图像压缩模型进行训练。
108.所述损失函数包括：
109.l＝r+λd＝i+h+λd；
110.其中，r为编码码率，编码码率r包含图像的编码码率ri和超先验的编码码率rh，d为图像失真，λ表示编码码率r和图像失真d的权衡，λ越高，以编码码率的增加为代价换取失真越小，反之亦然。
111.此外，在训练过程中，选择adam优化器，设置初始学习率为0.0001，在非局部块中，设置通道下采样倍数s和分辨率下采样倍数d为4。
112.进一步的，在得到目标火星图像压缩模型之后，还包括：获取验证数据集，并基于所述验证数据集对所述目标火星图像压缩模型的超参数进行调整，以提高所述目标火星图像压缩模型的准确率，即提高目标火星图像压缩模型的泛化能力。
113.更进一步的，在基于验证数据集获取到最优的目标火星图像压缩模型之后，还可通过测试数据集对所述目标火星图像压缩模型的准确性进行测试。
114.可选的，在得到目标火星图像压缩模型后，本技术实施例还通过以下几种示例性方法验证了目标火星图像压缩模型对待压缩火星图像进行压缩的有效性。
115.示例一、通过对比试验验证目标火星图像压缩模型对待压缩火星图像进行压缩的有效性：获取测试集中的火星图像，基于目标火星图像压缩模型和hevc算法对测试集中的火星图像进行压缩。分别获取目标火星图像压缩模型和现有算法(hevc)对测试集中的火星图像进行压缩的“编码码率-失真”曲线。以psnr为指标的“编码码率-失真”曲线如图5a所示，曲线501为目标火星图像压缩模型对应的“编码码率-失真”曲线，曲线502为现有算法对应的“编码码率-失真”曲线；以ms-ssim为指标的“编码码率-失真”曲线如图5b所示，曲线503为目标火星图像压缩模型对应的“编码码率-失真”曲线，曲线504为现有算法对应的“编码码率-失真”曲线。其中，横坐标表示码率(比特深度)。显然，目标火星图像压缩模型实现了最好的编码效果。
116.本实施例通过采用不同的火星图像压缩方法对测试集中的火星图像进行压缩，以jpeg为基准，计算bd-psnr(db)，bd-msssim以及对应的bd-rate(％)，结果如下表1所示。
117.表1
[0118][0119]
其中，bd-psnr(db)和bd-msssim指标越高，表明对火星图像的压缩效果越好；bd-rate(％)指标越低，对火星图像的压缩效果越好。
[0120]
示例二、通过消融实验对目标火星图像压缩模型的有效性进行验证。具体的，将非局部模块从目标火星图像压缩模型中去除，利用得到的新模型对测试集中的火星图像进行压缩，将压缩效果与目标火星图像压缩模型对该火星图像进行压缩的压缩效果进行对比。进一步表明目标火星图像压缩模型具有更高的编码性能，即可对火星图像进行高效压缩。
[0121]
基于同一发明构思，作为对上述方法的实现，本技术实施例还提供了一种火星图像压缩装置，该装置实施例与前述方法实施例对应，为便于阅读，本装置实施例不再对前述方法实施例中的细节内容进行逐一赘述，但应当明确，本实施例中的火星图像压缩装置能够对应实现前述方法实施例中的全部内容。
[0122]
图6为本技术一个实施例提供的火星图像压缩装置的结构框图，如图6所示，本实施例提供的轨道交通站点播报装置600包括：
[0123]
采集模块610，用于获取待压缩火星图像；
[0124]
判断模块620，用于根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩；其中，所述目标火星图像压缩模型包括第一编码器、第一解码器、第二编码器、以及第二解码器，其中，所述第一编码器、所述第一解码器、所述第二编码器、以及所述第二解码器均包括一个或多个非局部模块，所述非局部模块用于提取所述待压缩火星图像的非局部特征；
[0125]
处理模块630，用于将所述待压缩火星图像输入所述目标火星图像压缩模型，获取所述待压缩火星图像对应的压缩图像。
[0126]
在本技术一些实施例中，所述非局部模块包括非局部块、多个残差块，以及一个或多个卷积层；其中，所述非局部块与多个残差块连接，残差块之后连接一个卷积层，所述非局部块包括多个下采样单元和多个最大池化层。
[0127]
图7为本技术另一个实施例提供的火星图像压缩装置的结构框图，如图7所示，在图6所示装置的基础上，还包括计算模块710；在根据所述待压缩火星图像的非局部相似度与预设相似度阈值的关系，判断是否利用目标火星图像压缩模型对所述待压缩火星图像进行压缩之前，用于将所述待压缩火星图像分割为多个第一图像块，所述多个第一图像块互不重叠；确定所述多个第一图像块之间的结构相似性或均方根误差，并根据所述结构相似性或所述均方根误差确定所述待压缩火星图像的非局部相似度与预设相似度阈值的关系。
[0128]
在本技术一些实施例中，所述计算模块710还用于对所述待压缩火星图像加入加性高斯噪声，得到加噪火星图像；将所述加噪火星图像分割为多个第二图像块，并将所述第二图像块中相似的图像块匹配在一起进行滤波处理，得到去噪火星图像；获取所述待压缩
火星图像和所述去噪火星图像的峰值信噪比，根据所述峰值信噪比确定所述待压缩火星图像的非局部相似度与预设相似度阈值的关系。
[0129]
在本技术一些实施例中，所述装置还包括构建模块720，用于在将所述待压缩火星图像输入目标火星图像压缩模型进行压缩之前，获取所述目标火星图像压缩模型；其中，所述获取所述目标火星图像压缩模型包括：
[0130]
在超先验模型中插入一个或多个非局部模块，构建初始火星图像压缩模型；根据目标样本数据集中图像的非局部相似度与所述预设相似度阈值的关系获取训练样本集，并基于所述训练样本集对所述初始火星图像压缩模型进行训练，得到所述目标火星图像压缩模型。
[0131]
在本技术一些实施例中，所述构建模块720，具体用于将所述训练图像输入所述初始火星图像压缩模型，获取所述训练图像对应的压缩图像；根据所述训练图像对应的压缩图像的失真和编码码率对初始火星图像压缩模型的权重参数进行调整，直至预设损失函数收敛，得到所述目标火星图像压缩模型。
[0132]
在本技术一些实施例中，所述装置还包括验证模块730，用于获取验证数据集，并基于所述验证数据集对所述目标火星图像压缩模型的超参数进行调整，以提高所述目标火星图像压缩模型的准确率。
[0133]
本实施例提供的火星图像压缩装置可以执行上述方法实施例提供的轨道交通站点播报方法，其实现原理与技术效果类似，此处不再赘述。
[0134]
关于火星图像压缩装置的具体限定可以参见上文中对于火星图像压缩方法的限定，在此不再赘述。上述火星图像压缩装置中的各个模块可全部或部分通过软件、硬件及其组合来实现。上述各模块可以硬件形式内嵌于或独立于电子设备中的处理器中，也可以以软件形式存储于电子设备中的存储器中，以便于处理器调用执行以上各个模块对应的操作。
[0135]
在一个实施例中，提供了一种计算机设备，该计算机设备可以是终端，其内部结构图可以如图8所示。该计算机设备包括通过系统总线连接的处理器、存储器、通信接口、显示屏和输入装置。其中，该计算机设备的处理器用于提供计算和控制能力。该计算机设备的存储器包括非易失性存储介质、内存储器。该非易失性存储介质存储有操作系统和计算机程序。该内存储器为非易失性存储介质中的操作系统和计算机程序的运行提供环境。该计算机设备的通信接口用于与外部的终端进行有线或无线方式的通信，无线方式可通过wifi、运营商网络、近场通信(nfc)或其他技术实现。该计算机程序被处理器执行时以实现一种火星图像压缩方法。该计算机设备的显示屏可以是液晶显示屏或者电子墨水显示屏，该计算机设备的输入装置可以是显示屏上覆盖的触摸层，也可以是计算机设备外壳上设置的按键、轨迹球或触控板，还可以是外接的键盘、触控板或鼠标等。
[0136]
本领域技术人员可以理解，图8中示出的结构，仅仅是与本技术方案相关的部分结构的框图，并不构成对本技术方案所应用于其上的计算机设备的限定，具体的计算机设备可以包括比图中所示更多或更少的部件，或者组合某些部件，或者具有不同的部件布置。
[0137]
在一个实施例中，本技术提供的火星图像压缩方法可以实现为一种计算机程序的形式，计算机程序可在如图8所示的计算机设备上运行。计算机设备的存储器中可存储组成该火星图像压缩装置的各个程序模块，比如，图6所示的判断模块620、处理模块630。各个程
序模块构成的计算机程序使得处理器执行本说明书中描述的本技术各个实施例的火星图像压缩方法中的步骤。
[0138]
在一个实施例中，提供了一种电子设备，包括存储器和处理器，该存储器存储有计算机程序，该处理器执行计算机程序时实现本技术任一实施例所述的火星图像压缩方法。
[0139]
本实施例提供的电子设备，可以实现上述方法实施例提供的火星图像压缩方法，其实现原理与技术效果类似，此处不再赘述。
[0140]
在一个实施例中，提供了一种计算机可读存储介质，其上存储有计算机程序，计算机程序被处理器执行时实现本技术任一实施例所述的火星图像压缩方法。
[0141]
本实施例提供的计算机可读存储介质上存储的计算机程序，可以实现上述方法实施例提供的火星图像压缩方法，其实现原理与技术效果类似，此处不再赘述。
[0142]
本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程，是可以通过计算机程序来指令相关的硬件来完成，计算机程序可存储于一非易失性计算机可读取存储介质中，该计算机程序在执行时，可包括如上述各方法的实施例的流程。其中，本技术所提供的各实施例中所使用的对存储器、数据库或其它介质的任何引用，均可包括非易失性和易失性存储器中的至少一种。非易失性存储器可包括只读存储器(read-only memory，rom)、磁带、软盘、闪存或光存储器等。易失性存储器可包括随机存取存储器(random access memory，ram)或者外部高速缓冲存储器。作为说明而非局限，ram以多种形式可得，比如静态随机存取存储器(static random access memory，sram)和动态随机存取存储器(dynamic random access memory，dram)等。
[0143]
以上实施例的各技术特征可以进行任意的组合，为使描述简洁，未对上述实施例中的各个技术特征所有可能的组合都进行描述，然而，只要这些技术特征的组合不存在矛盾，都应当认为是本说明书记载的范围。
[0144]
以上实施例仅表达了本技术的几种实施方式，其描述较为具体和详细，但并不能因此而理解为对发明专利范围的限制。应当指出的是，对于本领域的普通技术人员来说，在不脱离本技术构思的前提下，还可以做出若干变形和改进，这些都属于本技术的保护范围。因此，本技术专利的保护范围应以所附权利要求为准。