模型训练制造技术

本说明书公开了模型训练

【技术实现步骤摘要】
模型训练、图像生成、信息提取方法、装置及电子设备


[0001]本说明书涉及数字媒体
，尤其涉及模型训练
、
图像生成
、
信息提取方法
、
装置及电子设备
。

技术介绍

[0002]随着数字媒体技术以及人工智能技术的发展，图像数据的数字水印在各个社交媒体平台中受到广泛的应用，其中，数字水印技术是一种通过在图像数据中加入人眼不可见的水印信息来起到侵权识别
、
内容溯源等作用的技术，以包含用户的私有数据
。
[0003]但是，由于在实际应用场景中，图像数据除了通过互联网直接进行传播外，还可能会在打印
、
拍摄后进行传播，而针对打印
、
拍摄后的图像数据，很难有效地从中再次提取出预先加入的水印信息，从而导致加入水印信息的图像数据经过打印
、
拍摄后，难以在传播过程中实现侵权识别和内容溯源
。

技术实现思路

[0004]本说明书提供一种模型训练
、
图像生成
、
信息提取方法
、
装置及电子设备，以部分的解决现有技术存在的图像数据中包含的水印信息难以有效地提取的问题
。
[0005]本说明书采用下述技术方案：
[0006]本说明书提供了一种模型训练方法，包括：
[0007]获取样本图像；
[0008]将所述样本图像输入到预设的水印图像生成模型中，以通过所述水印图像生成模型将预设的第一水印信息添加到所述样本图像中，得到目标样本图像；
[0009]按照指定传播方式，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像；
[0010]将所述目标样本图像对应的传播后图像输入到预设的水印提取模型中，以通过所述水印提取模型提取出所述传播后图像中包含的水印信息，作为第二水印信息；
[0011]以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，以及，对所述水印提取模型进行训练，得到训练后水印提取模型
。
[0012]可选地，所述水印图像生成模型包括：图像编码器
、
图像解码器；
[0013]将所述样本图像输入到所述水印图像生成模型中，通过所述水印图像生成模型将预设的第一水印信息添加到所述样本图像中，得到目标样本图像，具体包括：
[0014]将所述样本图像输入到所述图像编码器中，以通过所述图像编码器对所述样本图像进行编码，得到所述样本图像的特征图，作为第一特征图，并将确定出的所述第一水印信息对应的水印编码与所述第一特征图进行拼接，得到所述样本图像对应的第二特征图；
[0015]将所述第二特征图输入到所述图像解码器中，以通过所述图像解码器，得到目标样本图像
。
[0016]可选地，通过所述图像解码器得到目标样本图像，具体包括：
[0017]通过所述图像解码器，根据所述第二特征图生成基础目标样本图像；
[0018]针对所述基础目标样本图像中包含的每个像素，从所述样本图像中包含的各像素中确定出与该像素的位置相对应的像素，作为该像素对应的原始像素；
[0019]将该像素的像素值与该像素对应的原始像素的像素值进行残差连接，得到该像素的目标像素值；
[0020]根据所述基础目标样本图像中包含的各像素的目标像素值，得到目标样本图像
。
[0021]可选地，按照指定传播方式，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像，具体包括：
[0022]将所述目标样本图像输入到预先训练的图像转换模型中，以通过所述图像转换模型确定所述目标样本图像中的每个像素的像素值在按照指定传播方式进行传播后的变化值，并根据所述变化值，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像
。
[0023]可选地，训练所述图像转换模型，具体包括：
[0024]获取历史图像以及所述历史图像在经过指定传播方式传播后的目标传播图像；
[0025]将所述历史图像输入到所述图像转换模型中，以通过所述图像转换模型确定所述历史图像中的每个像素的像素值在经过指定传播方式进行传播后的变化值，并根据所述变化值得到所述历史图像对应的传播后图像；
[0026]以最小化所述历史图像对应的传播后图像和所述目标传播图像之间的偏差为优化目标，对所述图像转换模型进行训练，得到训练后图像转换模型
。
[0027]可选地，以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，具体包括：
[0028]以最小化所述第一水印信息和所述第二水印信息之间的偏差和最小化所述目标样本图像和所述样本图像之间的偏差为优化目标，对所述水印图像生成模型进行训练
。
[0029]可选地，以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，具体包括：
[0030]将所述样本图像以及所述目标样本图像作为待判别图像集；
[0031]针对所述待判别图像集中包含的每个图像，将该图像输入到预设的判别模型中，以通过所述判别模型确定用于表示该图像是否为添加水印信息的图像的判别结果；
[0032]以最小化所述判别结果与该图像是否为添加水印信息的图像的实际结果之间的偏差和最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，至少对所述水印图像生成模型进行训练
。
[0033]本说明书提供了一种图像生成方法，包括：
[0034]获取基础图像；
[0035]将所述基础图像输入到预先训练的水印图像生成模型中，以通过所述水印图像生成模型将预设的水印信息添加到所述基础图像中，得到目标图像，所述水印图像生成模型是通过上述模型训练方法训练得到；
[0036]通过所述目标图像进行任务执行
。
[0037]本说明书提供了一种信息提取方法，包括：
[0038]获取目标图像，所述目标图像是通过预先训练的水印图像生成模型在基础图像中加入预设的水印信息后所得到的图像，所述水印图像生成模型是通过上述模型训练方法训练得到；
[0039]将所述目标图像输入到预先训练的水印提取模型中，以通过所述水印提取模型，从所述目标图像中提取出所述目标图像中包含的水印信息，所述水印提取模型是通过上述模型训练方法训练得到；
[0040]根据所述水印信息，执行任务
。
[0041]本说明书提供了一种模型训练装置，包括：
[0042]获取模块，用于获取样本图像；
[0043]生成模块，用于将所述样本图像输入到预设的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种模型训练方法，包括：获取样本图像；将所述样本图像输入到预设的水印图像生成模型中，以通过所述水印图像生成模型将预设的第一水印信息添加到所述样本图像中，得到目标样本图像；按照指定传播方式，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像；将所述目标样本图像对应的传播后图像输入到预设的水印提取模型中，以通过所述水印提取模型提取出所述传播后图像中包含的水印信息，作为第二水印信息；以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，以及，对所述水印提取模型进行训练，得到训练后水印提取模型
。2.
如权利要求1所述的方法，所述水印图像生成模型包括：图像编码器
、
图像解码器；将所述样本图像输入到所述水印图像生成模型中，通过所述水印图像生成模型将预设的第一水印信息添加到所述样本图像中，得到目标样本图像，具体包括：将所述样本图像输入到所述图像编码器中，以通过所述图像编码器对所述样本图像进行编码，得到所述样本图像的特征图，作为第一特征图，并将确定出的所述第一水印信息对应的水印编码与所述第一特征图进行拼接，得到所述样本图像对应的第二特征图；将所述第二特征图输入到所述图像解码器中，以通过所述图像解码器，得到目标样本图像
。3.
如权利要求2所述的方法，通过所述图像解码器得到目标样本图像，具体包括：通过所述图像解码器，根据所述第二特征图生成基础目标样本图像；针对所述基础目标样本图像中包含的每个像素，从所述样本图像中包含的各像素中确定出与该像素的位置相对应的像素，作为该像素对应的原始像素；将该像素的像素值与该像素对应的原始像素的像素值进行残差连接，得到该像素的目标像素值；根据所述基础目标样本图像中包含的各像素的目标像素值，得到目标样本图像
。4.
如权利要求1所述的方法，按照指定传播方式，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像，具体包括：将所述目标样本图像输入到预先训练的图像转换模型中，以通过所述图像转换模型确定所述目标样本图像中的每个像素的像素值在按照指定传播方式进行传播后的变化值，并根据所述变化值，对所述目标样本图像中至少部分像素的像素值进行调整，得到所述目标样本图像对应的传播后图像
。5.
如权利要求4所述的方法，训练所述图像转换模型，具体包括：获取历史图像以及所述历史图像在经过指定传播方式传播后的目标传播图像；将所述历史图像输入到所述图像转换模型中，以通过所述图像转换模型确定所述历史图像中的每个像素的像素值在经过指定传播方式进行传播后的变化值，并根据所述变化值得到所述历史图像对应的传播后图像；以最小化所述历史图像对应的传播后图像和所述目标传播图像之间的偏差为优化目标，对所述图像转换模型进行训练，得到训练后图像转换模型
。
6.
如权利要求1所述的方法，以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，具体包括：以最小化所述第一水印信息和所述第二水印信息之间的偏差和最小化所述目标样本图像和所述样本图像之间的偏差为优化目标，对所述水印图像生成模型进行训练
。7.
如权利要求1或6所述的方法，以最小化所述第一水印信息和所述第二水印信息之间的偏差为优化目标，对所述水印图像生成模型进行训练，得到训练后水印图像生成模型，具体包...

【专利技术属性】
技术研发人员：唐董琦，刘健，
申请(专利权)人：支付宝杭州信息技术有限公司，
类型：发明
国别省市：