适用于扩散模型的水印嵌入方法、装置、设备和存储介质

本申请实施例提供一种适用于扩散模型的水印嵌入方法、装置、设备和存储介质。应用于图像生成技术领域，该方法通过构建水印集合，并从所述水印集合中随机抽取水印信息输入预训练的扩散模型，所述水印集合中每个水印信息与用户标识一一对应；通过预设的微调模块对预训练的扩散模型的中间层参数进行调整，得到微调后的扩散模型，所述微调模块包括：卷积层、自适应平均池化层、第一全连接层、第二全连接层和第三全连接层；利用微调后的扩散模型将输入的水印信息嵌入至扩散模型的生成图像中，得到目标输出图像，所述目标输出图像包含水印信息。该方法在扩散模型生成图像的过程中嵌入水印信息，降低了计算成本，提高了水印嵌入的稳定性、可靠性。