基于分布对齐的扩散模型生成图像水印嵌入和提取方法

本发明公开了一种基于分布对齐的扩散模型生成图像水印嵌入和提取方法，包括：对特定分布进行随机采样作为水印并与随机噪声进行拼接得到初始向量；构建分布变换模型，将含水印初始向量转换为符合扩散模型要求分布的潜在表示，分布变换模型的损失函数训练模型；将变换模型输出的向量作为扩散模型的起始潜在表示生成含水印的图像完成水印的嵌入；最后利用扩散采样和分布变换模型的逆过程实现水印的提取。本发明设计了一个基于扩散模型对于潜在表示的分布要求和图像与潜在表示之间的联系的扩散模型生成图像水印嵌入和提取方法，能够实现较高的水印溯源率，解决扩散模型分发场景内容追溯准确率问题。