[发明专利]基于换底机制的陶瓷水印模型训练方法、装置和嵌密方法

权利要求书

1.一种基于换底机制的陶瓷水印模型训练方法，其特征在于，包括：

分别获取当前训练图像和水印信息；

将所述当前训练图像和所述水印信息输入到编码器中生成残差图像，将所述残差图像和所述当前训练图像相加得到第一水印图像；

计算所述第一水印图像与所述当前训练图像之间损失函数集合中各个损失函数的损失值，根据所述损失值对所述编码器进行调节，得到更新后的编码器；

判断所述编码器是否达到预设的换底条件；

当所述编码器达到所述换底条件时，对所述当前训练图像进行边缘提取，得到掩膜，利用所述掩膜和所述当前训练图像对所述第一水印图像进行换底，得到第二水印图像；

将所述第二水印图像放入预设的噪声层中进行噪声处理；

将经过噪声处理的第二水印图像送入解码器进行解码得到秘密信息，根据所述秘密信息和所述水印信息得到交叉熵损失函数损失值，根据所述交叉熵损失函数损失值对所述解码器进行更新，得到更新后的解码器；

利用所述掩膜和所述当前训练图像对所述第一水印图像进行换底，得到第二水印图像包括：

根据所述掩膜和所述当前训练图像得到所述当前训练图像的非边界图像；

根据所述掩膜和所述第一水印图像得到所述第一水印图像的边界图像；

将所述当前训练图像的非边界图像与所述第一水印图像的边界图像相加，得到所述第二水印图像。

2.根据权利要求1所述的方法，其特征在于，还包括：当所述编码器未达到所述换底条件时，将所述第一水印图像放入预设的噪声层中进行噪声处理；

将经过噪声处理的第一水印图像送入所述解码器进行解码得到秘密信息，根据所述秘密信息和所述水印信息得到交叉熵损失函数损失值，根据所述交叉熵损失函数损失值对所述解码器进行更新。

3.根据权利要求1或2所述的方法，其特征在于：

在根据所述损失值对所述编码器进行调节，得到更新后的编码器之前，还包括：获取所述损失函数集合中每个损失函数的权重值；

根据所述损失值对所述编码器进行调节，得到更新后的编码器包括：利用所述损失函数集合每个损失函数的损失值和对应的权重值对所述编码器进行调节，得到更新后的编码器；

在根据所述交叉熵损失函数损失值对所述解码器进行更新之前，还包括：获取所述交叉熵损失函数的权重值；

根据所述交叉熵损失函数损失值对所述解码器进行更新包括：利用所述交叉熵损失函数的权重值、所述交叉熵损失函数的损失值对所述解码器进行更新。

4.根据权利要求3所述的方法，其特征在于，所述损失函数集合中包括以下中的一项或几项：LPIPS损失函数、L无穷损失函数、CRITIC损失函数、MSE损失函数；

在所述换底条件之前且在预设的第一步数阈值之前，仅对交叉熵损失函数的权重值赋值；

在所述换底条件之前且在预设的第一步数阈值之后，所述LPIPS损失函数、所述CRITIC损失函数的权重值相同且比交叉熵损失函数的权重值要小，所述MSE损失函数的权重值小于所述LPIPS损失函数的权重值，所述L无穷损失函数权重值比LPIPS损失函数权重值小；

在所述换底条件之后，所述LPIPS损失函数、所述CRITIC损失函数的权重值相同且比交叉熵损失函数的权重值要小，所述MSE损失函数的权重值为0，所述L无穷损失函数权重值比LPIPS损失函数权重值小。

5.根据权利要求4所述的方法，其特征在于，当所述损失函数集合中包括MSE损失函数时，计算所述第一水印图像与所述当前训练图像之间所述MSE损失函数的损失值包括：

根据所述当前训练图像、所述第一水印图像和所述掩膜得到所述编码器中所述MSE损失函数的损失值。