[发明专利]基于生成嵌入概率图的隐藏信息检测方法和系统

权利要求书

1.一种基于生成嵌入概率图的隐藏信息检测方法，其特征在于，包括如下步骤：

S10，根据待测图像生成待测嵌入概率图，所述待测嵌入概率图记录所述待测图像中每个像素点携带隐藏秘密信息的概率，调整嵌入概率图生成模块，生成嵌入概率图，

采用交叉熵损失l_m和交并比损失IoU组合的总损失L_all优化嵌入概率图生成模块；其中总损失L_all计算公式为：

式(1)中，M为U²-Net的总深度，考虑到不同层对最后结果的贡献是具有差异性的，将不同层的交叉熵损失l_m赋予不同的权值，使得L_all更适用于细节处的刻画；其中l_m计算公式为：

其中(H,W)代表输入的图片的高和宽，(r,c)则代表其中各个像素点的值，P_G(r,c)和P_S(r,c)分别表示真实的嵌入概率图和模块生成的嵌入概率图的对应坐标像素值；

式(1)中，IoU损失计算方式为：

其中C_j和分别代表真实选框和模块所生成的嵌入概率图，式(3)用于约束生成图像的中的阴影区域的与真实图像之间的重合比；

S20，对待测图像和待测嵌入概率图进行随机翻转和旋转操作，以增强数据；

S30，对嵌入概率图生成模块进行剪枝操作，以去除高维语义信息干扰；

S40，将待测嵌入概率图和待测图像分别与多个高通滤波核进行卷积，得到待测嵌入概率图对应的第一残差图和待测图像对应的第二残差图；

S50，对第一残差图和第二残差图进行融合，得到待测融合图像，包括：采用Pseudo-Siamese结构对第一残差图和第二残差图进行融合，得到待测融合图像；将待测融合图像经过注意力机制处理后进行权重裁剪；

S60，采用预先训练得到的隐写分析模型对待测融合图像进行学习并输出所述待测图像是否藏有秘密信息的概率，以判断所述待测图像中是否隐藏秘密信息。

2.根据权利要求1所述的基于生成嵌入概率图的隐藏信息检测方法，其特征在于，隐写分析模型的训练过程包括：

采集多个样本图像，确定各个样本图像的样本标签；所述样本标签包括相应样本图像包括秘密信息的概率；

获取各个样本图像的融合图像，以各个样本图像的融合图像为输入，以各个样本图像的样本标签为输出训练初始神经网络模型，以得到隐写分析模型。

3.根据权利要求2所述的基于生成嵌入概率图的隐藏信息检测方法，其特征在于，获取各个样本图像的融合图像包括：

根据各个样本图像生成各个样本嵌入概率图；所述样本嵌入概率图记录相应样本图像中每个像素点携带隐藏秘密信息的概率；

将样本嵌入概率图和相应的样本图像分别与多个高通滤波核进行卷积，得到各个样本嵌入概率图对应的第三残差图和各个样本图像对应的第四残差图；

分别将各个第三残差图和相应的第四残差图进行融合，得到各个样本图像的融合图像。

4.根据权利要求1所述的基于生成嵌入概率图的隐藏信息检测方法的检测系统，其特征在于，包括：

生成模块，用于根据待测图像生成待测嵌入概率图；所述待测嵌入概率图记录所述待测图像中每个像素点携带隐藏秘密信息的概率；

卷积模块，用于将待测嵌入概率图和待测图像分别与多个高通滤波核进行卷积，得到待测嵌入概率图对应的第一残差图和待测图像对应的第二残差图；

融合模块，用于对第一残差图和第二残差图进行融合，得到待测融合图像；

输出模块，用于采用预先训练得到的隐写分析模型对待测融合图像进行学习并输出所述待测图像是否藏有秘密信息的概率，以判断所述待测图像中是否隐藏秘密信息。