摘要
本发明涉及一种文本生成图像模型的水印方法,包括:获取触发提示词、水印图像和文本生成图像模型,所述文本生成图像模型的组件包括文本编码器、图像自编码器和UNet;在所述文本编码器的预定义字典中添加触发提示词的Token索引;冻结所述文本生成图像模型中除文本编码器外的所有组件,通过自适应调整策略训练文本编码器中与触发提示词相关的参数;冻结所述文本编码器与图像自编码器,通过迭代训练调整UNet的参数使触发提示词与水印图像绑定;通过对抗训练提高水印鲁棒性,完成文本生成图像模型的水印嵌入。与现有技术相比,本发明可以准确实现文本生成图像模型的所有权认证。