一种针对图像数据的水印嵌入方法

专利主权项内容

1.一种针对图像数据的水印嵌入方法，其特征在于，包括：获取预设个数初始图像，对每个初始图像进行分割，得到所述每个初始图像对应的载体图像和水印图像，并将所述预设个数初始图像分为训练集和验证集；根据所述训练集中所述每个初始图像的所述载体图像和所述水印图像对预设水印网络进行融合和可微分模拟攻击的训练，得到目标水印网络；将所述验证集中所述每个初始图像的所述载体图像和所述水印图像输入至所述目标水印网络中，得到所述验证集中所有初始图像的整体损失数据集；当所述整体损失数据集中数值的和小于预设损失阈值时，确定所述目标水印网络训练完成；所述预设水印网络包括水印嵌入网络和水印提取网络；所述根据所述训练集中所述每个初始图像的所述载体图像和所述水印图像对预设水印网络进行融合和可微分模拟攻击的训练，得到目标水印网络，包括：从所述训练集中确定第一初始图像；将所述第一初始图像的所述载体图像和所述水印图像输入至所述水印嵌入网络中进行融合，得到初始水印载体图像和嵌入损失；对所述初始水印载体图像进行模拟攻击，得到攻击后的目标水印载体图像；根据所述嵌入损失、所述第一初始图像、所述目标水印载体图像和所述水印提取网络，得到整体损失；根据所述整体损失对所述预设水印网络的参数进行调整，得到初始水印网络；在所述整体损失达到预设损失阈值，或所述训练集中不存在未进行嵌入的初始图像的情况下，对训练过程中的参数进行调整，重新根据所述训练集对所述预设水印网络进行训练，得到目标水印网络。