一种信息隐藏图像处理模型构建方法、装置、终端及介质与流程

1.本技术涉及图像处理技术领域，尤其涉及一种信息隐藏图像处理模型构建方法、装置、终端及介质。


背景技术：

2.随着卷积神经网络的蓬勃发展，各种图像处理的处理技术都在深度学习的孕育下不断创新。其中，信息隐藏和图像转换在图像处理的多个领域得到广泛的应用。
3.信息隐藏又称为图像隐写，通过将载有秘密信息的秘密图像隐藏入与它尺寸相同的载体图像中，从而得到视觉效果与载体图像基本相似但又包含了秘密图像的载密图像。
4.图像转换是通过将特定图像的风格特征迁移到目标内容图中，使之在保持原内容信息的同时具有与目标图像一样的风格特征。
5.当前基于深度学习的可逆信息隐藏技术的编码部分与解码部分鲁棒性很差，图像转换过程中容易导致嵌入的秘密信息的破坏，使得载密图像经过图像转换难以从中再次解码到秘密信息，从而限制了图像转换技术和信息隐藏技术的结合。


技术实现要素：

6.本技术提供了一种信息隐藏图像处理模型构建方法、装置、终端及介质，用于解决现有基于深度学习的可逆信息隐藏技术的编码与解码鲁棒性差的技术问题。
7.本技术第一方面提供了一种信息隐藏图像处理模型构建方法，包括：
8.获取载体图像和秘密图像；
9.通过编码器网络，对所述载体图像和所述秘密图像进行信息隐藏编码处理，得到第一载密图像；
10.对所述第一载密图像与所述载体图像进行比较，以根据所述第一载密图像与所述载体图像的比较结果，确定载体图像损失；
11.通过图像转换网络，对所述第一载密图像进行图像转换处理，得到第二载密图像；
12.通过解码器网络，对所述第二载密图像进行隐藏信息解码处理，得到解密图像；
13.对所述解密图像与所述秘密图像进行比较，以根据所述解密图像与所述秘密图像的比较结果，确定秘密图像损失；
14.以图像总损失为优化目标，通过优化算法，对所述编码器网络和所述解码器网络进行优化，得到优化后的编码器网络和解码器网络，其中，所述图像总损失为根据所述载体图像损失和所述秘密图像损失计算得到的；
15.利用所述优化后的编码器网络、所述优化后的解码器网络以及所述图像转换网络，构建信息隐藏图像处理模型。
16.优选地，所述对所述第一载密图像与所述载体图像进行比较，以根据所述第一载密图像与所述载体图像的比较结果确定载体图像损失具体包括：
17.通过欧氏距离比较方式，对所述第一载密图像与所述载体图像进行比较，以根据
所述第一载密图像与所述载体图像的欧氏距离，确定载体图像损失。
18.优选地，所述对所述解密图像与所述秘密图像进行比较，以根据所述解密图像与所述秘密图像的比较结果，确定秘密图像损失具体包括：
19.通过欧氏距离比较方式，对所述解密图像与所述秘密图像进行比较，以根据所述解密图像与所述秘密图像的欧氏距离，确定秘密图像损失。
20.优选地，所述图像总损失计算过程具体包括：
21.计算所述载体图像损失和所述秘密图像损失的加权和，以所述加权和作为所述图像总损失。
22.优选地，所述图像总损失计算过程具体包括：
23.计算所述载体图像损失和所述秘密图像损失的和值，以所述和值作为所述图像总损失。
24.优选地，还包括：
25.对所述载体图像和/或所述秘密图像进行图像尺寸预处理，使得预处理后的所述载体图像和所述秘密图像尺寸一致。
26.优选地，所述优化算法具体为adam优化算法。
27.本技术第二方面提供了一种信息隐藏图像处理模型构建装置，包括：
28.图像获取单元，用于获取载体图像和秘密图像；
29.编码处理单元，用于通过编码器网络，对所述载体图像和所述秘密图像进行信息隐藏编码处理，得到第一载密图像；
30.载体图像损失确定单元，用于对所述第一载密图像与所述载体图像进行比较，以根据所述第一载密图像与所述载体图像的比较结果，确定载体图像损失；
31.图像转换单元，用于通过图像转换网络，对所述第一载密图像进行图像转换处理，得到第二载密图像；
32.解码处理单元，用于通过解码器网络，对所述第二载密图像进行隐藏信息解码处理，得到解密图像；
33.秘密图像损失确定单元，用于对所述解密图像与所述秘密图像进行比较，以根据所述解密图像与所述秘密图像的比较结果，确定秘密图像损失；
34.优化处理单元，用于以图像总损失为优化目标，通过优化算法，对所述编码器网络和所述解码器网络进行优化，得到优化后的编码器网络和解码器网络，其中，所述图像总损失为根据所述载体图像损失和所述秘密图像损失计算得到的；
35.模型构建单元，用于利用所述优化后的编码器网络、所述优化后的解码器网络以及所述图像转换网络，构建信息隐藏图像处理模型。
36.本技术第三方面提供了一种信息隐藏图像处理模型构建终端，包括：存储器和处理器；
37.所述存储器用于存储程序代码，所述程序代码与如本技术第一方面提供的信息隐藏图像处理模型构建方法相对应；
38.所述处理器用于执行所述程序代码。
39.本技术第四方面提供了一种计算机可读存储介质，所述计算机可读存储介质中保存有与如本技术第一方面提供的信息隐藏图像处理模型构建方法相对应的程序代码。
40.从以上技术方案可以看出，本技术具有以下优点：
41.基于本技术提供的方法通过图像转换处理前的载体图像损失和图像转换处理后的秘密图像损失，基于载体图像损失和秘密图像损失对编码器网络和解码器网络进行优化，加强了编码、图像转换以及解码三者间的耦合，提高了可逆信息隐藏技术的编码与解码鲁棒性，进而提高了从经过图像转换处理的载密图像中提取出秘密信息的成功率和准确率。
附图说明
42.为了更清楚地说明本技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其它的附图。
43.图1为本技术提供的一种信息隐藏图像处理模型构建方法的第一个实施例的流程示意图。
44.图2为本技术提供的一种信息隐藏图像处理模型构建方法的第二个实施例的流程示意图。
45.图3为本技术提供的一种信息隐藏图像处理模型构建装置的一个实施例的结构示意图。
46.图4为编码器网络、图像转网络和解码器网络的技术总框架示意图。
47.图5为本技术提供的信息隐藏图像处理模型构建方法的算法逻辑流程图。
48.图6为编码器网络和解码器网络的逻辑框图。
具体实施方式
49.本技术实施例提供了一种信息隐藏图像处理模型构建方法、装置、终端及介质，用于解决现有基于深度学习的可逆信息隐藏技术的编码与解码鲁棒性差的技术问题。
50.为使得本技术的发明目的、特征、优点能够更加的明显和易懂，下面将结合本技术实施例中的附图，对本技术实施例中的技术方案进行清楚、完整地描述，显然，下面所描述的实施例仅仅是本技术一部分实施例，而非全部的实施例。基于本技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其它实施例，都属于本技术保护的范围。
51.请参阅图1，本技术第一个实施例提供了一种信息隐藏图像处理模型构建方法，包括：
52.步骤101、获取载体图像和秘密图像。
53.步骤102、通过编码器网络，对载体图像和秘密图像进行信息隐藏编码处理，得到第一载密图像。
54.需要说明的是，首先，获取载体图像和秘密图像，通过编码器网络，基于获取到的载体图像和秘密图像进行信息隐藏编码处理，得到第一载密图像。
55.步骤103、对第一载密图像与载体图像进行比较，以根据第一载密图像与载体图像的比较结果，确定载体图像损失。
56.然后基于步骤102得到的第一载密图像与原始的载体图像进行比较，通过比较得到第一载密图像与载体图像的比较结果，以便根据该比较结果，确定载体图像损失。
57.步骤104、通过图像转换网络，对第一载密图像进行图像转换处理，得到第二载密图像。
58.基于步骤102获得的第一载密图像，利用图像转换网络对其进行图像转换的处理，从而得到图像转换后的载密图像，即本技术的第二载密图像。
59.步骤105、通过解码器网络，对第二载密图像进行隐藏信息解码处理，得到解密图像。
60.步骤106、对解密图像与秘密图像进行比较，以根据解密图像与秘密图像的比较结果，确定秘密图像损失。
61.基于步骤104得到的第二载密图像，通过解码器网络对其进行隐藏信息解码处理，从而得到相应的解密图像。
62.然后利用该解密图像与原始的秘密图像进行比较，以便根据得到的比较结果，确定解密图像与秘密图像之间的秘密图像损失。
63.步骤107、以图像总损失为优化目标，通过优化算法，对编码器网络和解码器网络进行优化，得到优化后的编码器网络和解码器网络。
64.其中，图像总损失为根据载体图像损失和秘密图像损失计算得到的。
65.步骤108、利用优化后的编码器网络、优化后的解码器网络以及图像转换网络，构建信息隐藏图像处理模型。
66.基于步骤103获得的载体图像损失和步骤106获得的秘密图像损失，计算出相应的图像总损失，然后以图像总损失为优化目标，通过优化算法，对编码器网络和解码器网络进行训练优化，再利用优化后的编码器网络和解码器网络构建出相应的信息隐藏图像处理模型，以便利用该模型的编码器网络和解码器网络，对待处理的载体图像和秘密图像进行信息隐藏编解码处理。
67.本技术实施例提供的方法基于载体图像损失和秘密图像损失对编码器网络和解码器网络进行优化，加强了编码、图像转换以及解码三者间的耦合，提高了可逆信息隐藏技术的编码与解码鲁棒性，进而提高了从经过图像转换处理的载密图像中提取出秘密信息的成功率和准确率。
68.以上为本技术提供的一种信息隐藏图像处理模型构建方法的第一个实施例的详细说明，下面为本技术在第一个实施例的基础上所提供的一种信息隐藏图像处理模型构建方法的第二个实施例的详细说明。
69.请参阅图2，本技术第二个实施例提供的一种信息隐藏图像处理模型构建方法，包括：
70.在第一个实施例的基础上，更具体地，在本实施例中，实施例一中的步骤103可以进一步细化为：
71.步骤103、通过欧氏距离比较方式，对第一载密图像与载体图像进行比较，以根据第一载密图像与载体图像的欧氏距离，确定载体图像损失。
72.更具体地，实施例一当中的步骤106可以进一步细化为：
73.步骤106、通过欧氏距离比较方式，对解密图像与秘密图像进行比较，以根据解密
图像与秘密图像的欧氏距离，确定秘密图像损失。
74.更具体地，图像总损失计算过程具体包括：
75.计算载体图像损失和秘密图像损失的加权和，以加权和作为图像总损失。
76.更具体地，图像总损失计算过程具体包括：
77.计算载体图像损失和秘密图像损失的和值，以和值作为图像总损失。
78.更具体地，还包括：
79.步骤100、对载体图像和/或秘密图像进行图像尺寸预处理，使得预处理后的载体图像和秘密图像尺寸一致。
80.更具体地，优化算法具体为adam优化算法。
81.本技术目的在于搭建总网络模型，即信息隐藏图像处理模型，该网络模型主要由编码器网络、图像转网络和解码器网络构成，技术总框架流程如图4所示；
82.如图5所示，分别构造一个将秘密图像嵌入载体图像的编码器网络和将秘密图像从载密图像中解码出来的解码器网络；
83.其中，编码器网络hnet和解码器网络rnet分别由若干个卷积层构成，具体如图6所示；
84.将载体图像和秘密图像集依次输入编码器网络，获得载密图像；然后根据载体图像和载密图像计算载体图像损失：此过程具体包括：
85.(1)将秘密图像s与载体图像c都处理成统一的大小，优选为224
×
224；
86.(2)将秘密图像s和载体图像c输入到编码器网络hnet中，前向传播，得到视觉效果与载体图像c基本相似的载密图像stego；
87.(3)载体图像损失为载体图像c和载密图像stego的欧式距离，计算公式如下:
[0088][0089]
其中，c表示载体图像，stego表示第一载密图像，l(c,stego)为载体图像损失；
[0090]
将获得的载密图像输入图像转换模型中，得到载密图像的图像转换效果图；
[0091]
上述图像转换网络为当前任意图像转换技术对应的模型网络；
[0092]
将载密图像stego输入到上述图像转换模型中，获得载密图像stego的图像转换效果图stego'，即第二载密图像；
[0093]
将载密图像的图像转换效果图stego'输入解码器网络rnet中，得到解密图像s'，再由原秘密图像s和解密图像s'计算秘密图像损失l(s,s')，秘密图像损失为原秘密图像s和解密图像s'的欧式距离，计算公式如下：
[0094][0095]
将上述载密图像损失l(c,stego)和秘密图像损失l(s,s')的加权和作为模型的图像总损失l
total
,总损失计算公式如下：
[0096]
l
total
＝αl(c,stego) βl(s,s')
ꢀꢀ
(3)
[0097]
其中，α和β为常数，分别代表封面损失l(c,stego)和秘密损失l(s,s')的权重，此外除了加权后外，也可以采用载密图像损失与秘密图像损失之和作为图像总损失；
[0098]
然后的基于图像总损失进行优化训练，通过从数据集中随机选取53228张图像，把其中26614张作为载体图像数据集，剩余图像作为秘密图像数据集；
[0099]
利用载体图像数据集和秘密图像数据集的图像，重复训练优化编码器网络hnet和解码器网络rnet，训练中，学习率可以设置为1
×
103，迭代次数为100次，使用adam优化算法优化网络；
[0100]
经过100次迭代训练后，当模型的总损失收敛并且为最小值，编码器网络hnet编码得到的载密图像stego与载体图像c在视觉上基本无差别，而且经过解码器网络rnet获取解密图像s'与初始的秘密图像s相似度高。
[0101]
通过替换不同的图像转换网络，可以实现将信息隐藏技术与任意图像转换技术结合，提高了技术的灵活度和应用性。
[0102]
以上为本技术提供的一种信息隐藏图像处理模型构建方法的第二个实施例的详细说明，下面为本技术提供的一种信息隐藏图像处理模型构建装置的一个实施例的详细说明。
[0103]
请参阅图3，本技术第三个实施例提供了一种信息隐藏图像处理模型构建装置，包括：
[0104]
图像获取单元201，用于获取载体图像和秘密图像；
[0105]
编码处理单元202，用于通过编码器网络，对载体图像和秘密图像进行信息隐藏编码处理，得到第一载密图像；
[0106]
载体图像损失确定单元203，用于对第一载密图像与载体图像进行比较，以根据第一载密图像与载体图像的比较结果，确定载体图像损失；
[0107]
图像转换单元204，用于通过图像转换网络，对第一载密图像进行图像转换处理，得到第二载密图像；
[0108]
解码处理单元205，用于通过解码器网络，对第二载密图像进行隐藏信息解码处理，得到解密图像；
[0109]
秘密图像损失确定单元206，用于对解密图像与秘密图像进行比较，以根据解密图像与秘密图像的比较结果，确定秘密图像损失；
[0110]
优化处理单元207，用于以图像总损失为优化目标，通过优化算法，对编码器网络和解码器网络进行优化，得到优化后的编码器网络和解码器网络，其中，图像总损失为根据载体图像损失和秘密图像损失计算得到的；
[0111]
模型构建单元208，用于利用优化后的编码器网络、优化后的解码器网络以及图像转换网络，构建信息隐藏图像处理模型。
[0112]
另外，本技术第四个实施例提供了一种信息隐藏图像处理模型构建终端，包括：存储器和处理器；
[0113]
存储器用于存储程序代码，程序代码与如本技术第一个实施例或第二个实施例提供的信息隐藏图像处理模型构建方法相对应；
[0114]
处理器用于执行程序代码。
[0115]
本技术第五个实施例提供了一种计算机可读存储介质，计算机可读存储介质中保存有与如本技术第一个实施例或第二个实施例提供的信息隐藏图像处理模型构建方法相对应的程序代码。
[0116]
所属领域的技术人员可以清楚地了解到，为描述的方便和简洁，上述描述的终端，装置和单元的具体工作过程，可以参考前述方法实施例中的对应过程，在此不再赘述。
[0117]
在本技术所提供的几个实施例中，应该理解到，所揭露的终端，装置和方法，可以通过其它的方式实现。例如，以上所描述的装置实施例仅仅是示意性的，例如，所述单元的划分，仅仅为一种逻辑功能划分，实际实现时可以有另外的划分方式，例如多个单元或组件可以结合或者可以集成到另一个系统，或一些特征可以忽略，或不执行。另一点，所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口，装置或单元的间接耦合或通信连接，可以是电性，机械或其它的形式。
[0118]
本技术的说明书及上述附图中的术语“第一”、“第二”、“第三”、“第四”等(如果存在)是用于区别类似的对象，而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换，以便这里描述的本技术的实施例，例如能够以除了在这里图示或描述的那些以外的顺序实施。此外，术语“包括”和“具有”以及他们的任何变形，意图在于覆盖不排他的包含，例如，包含了一系列步骤或单元的过程、方法、系统、产品或设备不必限于清楚地列出的那些步骤或单元，而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。
[0119]
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的，作为单元显示的部件可以是或者也可以不是物理单元，即可以位于一个地方，或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0120]
另外，在本发明各个实施例中的各功能单元可以集成在一个处理单元中，也可以是各个单元单独物理存在，也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现，也可以采用软件功能单元的形式实现。
[0121]
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的全部或部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括：u盘、移动硬盘、只读存储器(rom，read
‑
only memory)、随机存取存储器(ram，random access memory)、磁碟或者光盘等各种可以存储程序代码的介质。
[0122]
以上所述，以上实施例仅用以说明本技术的技术方案，而非对其限制；尽管参照前述实施例对本技术进行了详细的说明，本领域的普通技术人员应当理解：其依然可以对前述各实施例所记载的技术方案进行修改，或者对其中部分技术特征进行等同替换；而这些修改或者替换，并不使相应技术方案的本质脱离本技术各实施例技术方案的精神和范围。