面向稳定扩散模型的图像水印方法

本发明公开了一种面向稳定扩散模型的图像水印方法，从稳定扩散模型的逆向扩散过程中所有潜在特征选择一个作为目标潜在特征，构建包括水印信息嵌入模块和水印提取模块的图像水印生成识别模型，水印信息嵌入模块用于将用户水印信息嵌入目标潜在特征中，将得到的水印目标潜在特征代替原有目标潜在特征输入稳定扩散模型的剩余子模型生成水印图像；水印信息提取模块用于从对水印图像进行DDIM反转得到的水印潜在特征中提取用户水印信息，根据提取出的用户水印信息即可识别水印图像的生成用户。本发明通过在稳定扩散模型的潜在特征中嵌入用户水印信息，在不破坏生成图像质量的前提下实现高鲁棒性、强隐蔽性的图像溯源能力。