[发明专利]基于Transformer的软融合鲁棒图像水印方法

权利要求书

1.基于Transformer的软融合鲁棒图像水印方法，其特征在于，包括：

构建基于Transformer的端到端水印模型；其中，所述端到端水印模型包括：编码器、解码器、噪声层和鉴别器；

将原始图像和原始水印输入所述编码器进行编码，获取编码图像，完成水印融合；

将所述编码图像输入所述噪声层进行噪声添加，获取噪声图像；

将所述噪声图像输入所述解码器进行水印提取，获取解码后的水印；

基于所述鉴别器，对所述编码图像进行鉴别；

所述编码器包括：水印预处理模块、第一卷积层、特征增强模块、软融合模块、Transformer块和第二卷积层；

所述原始水印输入所述水印预处理模块进行扩展，获取初始水印特征图，所述原始图像输入所述第一卷积层进行卷积处理，获取初始图像特征图；

将所述初始水印特征图和所述初始图像特征图输入所述特征增强模块，获取增强特征图；

将所述增强特征图和所述初始水印特征图输入所述软融合模块，获取带水印特征图；

将所述带水印特征图通过Transformer块后与所述原始图像进行连接，再通过所述第二卷积层进行卷积处理，获取含水印的所述编码图像；

通过所述软融合模块获取所述带水印特征图的过程表示为：

其中，Attention_w和Attention_c分别表示初始水印和原始图像的注意力图，P(.) 为线性映射，为特征拼接操作，为增强特征图生成的2D 查询张量，F_wc为带水印特征图。

2.根据权利要求1所述的基于Transformer的软融合鲁棒图像水印方法，其特征在于，所述编码器的编码过程表示为：其中，I_c为原始图像，I_e为编码图像，E为编码器，W为原始水印，θ_E为编码器的参数，L_E为编码损失。

3.根据权利要求1所述的基于Transformer的软融合鲁棒图像水印方法，其特征在于，所述解码器进行水印提取的过程表示为：其中，W为原始水印，W^，为解码器提取的水印，I_n为噪声图像，θ_D为解码器的参数，D为解码器，为解码损失。