一种基于Swin-Transformer的可逆盲水印鲁棒性提升方法

本申请涉及基于Swin‑Transformer的可逆盲水印鲁棒性提升方法，包括：用Swin‑Transformer提取输入图像的多尺度特征；根据多尺度特征生成可逆盲水印，将水印嵌入至输入图像；利用动态噪声层选择噪声添加至输入图像，生成失真图像；利用Swin‑Transformer提取失真图像的多尺度特征，使用解码器恢复嵌入的可逆水印；根据可逆水印对多尺度特征进行特征恢复；利用编码器对特征恢复的特征进行水印的自适应特征重编码并迭代设定次数；在损失函数值大于前一次迭代的损失函数值时，调整水印的嵌入参数、特征恢复参数和失真图像的多尺度特征的融合参数。通过动态噪声层设计，结合Swin‑Transformer，能够有效应对多种失真类型，增强水印的鲁棒性，使得在几何失真与非几何失真下，水印的提取精度较现有技术提高。