基于深度学习的电脑屏幕最佳水印类型确定方法及系统

摘要

本发明涉及一种基于深度学习的电脑屏幕最佳水印类型确定方法及系统。该方法包括：对不同型号的电脑屏幕添加不同水印强度的水印类型，获取电脑屏幕图像数据集；确定电脑屏幕图像的视觉效果等级；以电脑屏幕图像为输入，以电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型；对当前电脑屏幕添加不同水印强度的水印类型，获取实际电脑屏幕图像；将实际电脑屏幕图像输入深度学习网络模型中，输出实际电脑屏幕图像的视觉效果等级；根据实际电脑屏幕图像的视觉效果等级筛选出最佳视觉效果等级，并将最佳视觉效果等级所对应的水印类型作为对当前电脑屏幕添加的最佳水印类型。本发明能够准确选择出最合适的水印类型以使得电脑屏幕的视觉效果最佳。

说明书

技术领域

本发明涉及电脑屏幕数字水印视觉效果领域，特别是涉及一种基于深度学习的电脑屏幕最佳水印类型确定方法及系统。

背景技术

当前，数字水印技术主要用于信息安全，大体分为明水印以及隐形水印，明水印显然是直接影响了视觉效果，虽然传统隐形水印的视觉效果较好，但其鲁棒性不足，水印容易被攻击破坏。屏幕水印本质上也是一种数字水印，屏幕水印的视觉效果受水印强度影响，水印强度越强，其抗攻击性，抗干扰能力越强，溯源提取能力也就越强，但视觉效果会变差。因此，为了满足客户的实际需求，衡量屏幕水印的溯源成功率与视觉效果之间的关系具有较大的应用背景和意义。

为保证信息安全，在电脑屏幕上添加数字水印，计算机的电子数据得到相对的安全保障，防止拍屏、截屏泄密，帮助用户精准定位泄密源头，追查责任人，但添加数字水印后，对正常办公会造成视觉干扰，影响体验效果。现有的电脑屏幕添加数字水印的方法均是将电脑中所存储的固有数字水印添加电脑屏幕，由人工观测添加数字水印电脑屏幕的视觉效果以决定选择哪种水印类型，但这种视觉效果观测方式受人工的主观意识强烈，无法在保障电脑屏幕信息安全的同时，选择出最合适的水印类型以使得电脑屏幕的视觉效果最佳。

发明内容

本发明的目的是提供一种基于深度学习的电脑屏幕最佳水印类型确定方法及系统，以解决由人工观测添加数字水印电脑屏幕的视觉效果的方式无法在保障电脑屏幕信息安全的同时，选择出最合适的水印类型以使得电脑屏幕的视觉效果最佳的问题。

为实现上述目的，本发明提供了如下方案：

一种基于深度学习的电脑屏幕最佳水印类型确定方法，包括：

对不同型号的电脑屏幕添加不同水印强度的水印类型，获取电脑屏幕图像数据集；所述电脑屏幕图像数据集包括多张电脑屏幕图像；

对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级；

以所述电脑屏幕图像为输入，以所述电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型；

对当前电脑屏幕添加不同水印强度的水印类型，获取实际电脑屏幕图像；

将所述实际电脑屏幕图像输入所述深度学习网络模型中，输出所述实际电脑屏幕图像的视觉效果等级；

根据所述实际电脑屏幕图像的视觉效果等级筛选出最佳视觉效果等级，并将所述最佳视觉效果等级所对应的水印类型作为对所述当前电脑屏幕添加的最佳水印类型；以所述最佳水印类型对应的水印强度值对所述当前电脑屏幕添加数字水印。

可选的，所述对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级，具体包括：

计算每张所述电脑屏幕图像的每一类水印类型不同水印强度的视觉信息保真度；

根据所述视觉信息保真度确定水印强度等级；

根据所述水印强度等级对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

可选的，所述以所述电脑屏幕图像为输入，以所述电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型，具体包括：

将所述电脑屏幕图像数据集内的电脑屏幕图像进行压缩，确定压缩后的电脑屏幕图像；

按照8:2的比例，将所述压缩后的电脑屏幕图像以及所述压缩后的电脑屏幕图像所对应的视觉效果等级随机划分为训练集和测试集；

以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型。

可选的，所述以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型，之后还包括：

采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型。

可选的，所述采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型，之后还包括：

以所述测试集内的压缩后的电脑屏幕图像为输入，以所述测试集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，验证所述修正后的深度学习网络模型。

一种基于深度学习的电脑屏幕最佳水印类型确定系统，包括：

电脑屏幕图像数据集获取模块，用于对不同型号的电脑屏幕添加不同水印强度的水印类型，获取电脑屏幕图像数据集；所述电脑屏幕图像数据集包括多张电脑屏幕图像；

视觉效果等级等级确定模块，用于对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级；

深度学习网络模型建立模块，用于以所述电脑屏幕图像为输入，以所述电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型；

实际电脑屏幕图像获取模块，用于对当前电脑屏幕添加不同水印强度的水印类型，获取实际电脑屏幕图像；

实际电脑屏幕图像的视觉效果等级输出模块，用于将所述实际电脑屏幕图像输入所述深度学习网络模型中，输出所述实际电脑屏幕图像的视觉效果等级；

最佳水印类型确定模块，用于根据所述实际电脑屏幕图像的视觉效果等级筛选出最佳视觉效果等级，并将所述最佳视觉效果等级所对应的水印类型作为对所述当前电脑屏幕添加的最佳水印类型；以所述最佳水印类型对应的水印强度值对所述当前电脑屏幕添加数字水印。

可选的，所述视觉效果等级等级确定模块，具体包括：

视觉信息保真度计算单元，用于计算每张所述电脑屏幕图像的每一类水印类型不同水印强度的视觉信息保真度；

水印强度等级确定单元，用于根据所述视觉信息保真度确定水印强度等级；

视觉效果等级确定单元，用于根据所述水印强度等级对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

可选的，所述深度学习网络模型建立模块，具体包括：

图像压缩单元，用于将所述电脑屏幕图像数据集内的电脑屏幕图像进行压缩，确定压缩后的电脑屏幕图像；

划分单元，用于按照8:2的比例，将所述压缩后的电脑屏幕图像以及所述压缩后的电脑屏幕图像所对应的视觉效果等级随机划分为训练集和测试集；

深度学习网络模型建立单元，用于以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型。

可选的，还包括：

修正单元，用于采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型。

可选的，还包括：

验证单元，用于以所述测试集内的压缩后的电脑屏幕图像为输入，以所述测试集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，验证所述修正后的深度学习网络模型。

根据本发明提供的具体实施例，本发明公开了以下技术效果：本发明提供了一种基于深度学习的电脑屏幕最佳水印类型确定方法及系统，通过对不同型号的电脑屏幕添加不同水印强度的水印类型的电脑屏幕图像数据集进行视觉效果等级，根据电脑屏幕图像以及视觉效果等级建立深度学习网络模型，从而根据该深度学习网络模型确定当前电脑屏幕添加的最佳水印类型，以达到最佳视觉效果，整个最佳水印类型选择过程无需人工参与，避免了人为主观意识的干扰，准确的选择出最合适的水印类型以使得电脑屏幕的视觉效果最佳，有效计算出电脑屏幕最佳的水印类型和水印强度值。

附图说明

为了更清楚地说明本发明实施例或现有技术中的技术方案，下面将对实施例中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其他的附图。

图1为本发明所提供的基于深度学习的电脑屏幕最佳水印类型确定方法流程图；

图2为本发明所提供的基于深度学习的电脑屏幕最佳水印类型确定系统结构图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

本发明的目的是提供一种基于深度学习的电脑屏幕最佳水印类型确定方法及系统，能够准确的选择出最合适的水印类型以使得电脑屏幕的视觉效果最佳，有效计算出电脑屏幕最佳的水印类型和水印强度值。

为使本发明的上述目的、特征和优点能够更加明显易懂，下面结合附图和具体实施方式对本发明作进一步详细的说明。

图1为本发明所提供的基于深度学习的电脑屏幕最佳水印类型确定方法流程图，如图1所示，一种基于深度学习的电脑屏幕最佳水印类型确定方法，包括：

步骤101：对不同型号的电脑屏幕添加不同水印强度的水印类型，获取电脑屏幕图像数据集；所述电脑屏幕图像数据集包括多张电脑屏幕图像。

在不同的电脑屏幕上安装不同类型的水印，设置不同的水印强度[0，1，2…10]，强度越大，视觉效果越差，对设置好水印的电脑屏幕进行截图、拍照，确定多张电脑屏幕图像；

步骤102：对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

所述步骤102具体包括：计算每张所述电脑屏幕图像的每一类水印类型不同水印强度的视觉信息保真度；根据所述视觉信息保真度确定水印强度等级；根据所述水印强度等级对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

步骤103：以所述电脑屏幕图像为输入，以所述电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型。

根据训练样本的规模选择合适的网络结构，由于屏幕截图的照片输入比较大，将样本缩放至320x240作为模型的输入。

所述步骤103具体包括：将所述电脑屏幕图像数据集内的电脑屏幕图像进行压缩，确定压缩后的电脑屏幕图像；按照8:2的比例，将所述压缩后的电脑屏幕图像以及所述压缩后的电脑屏幕图像所对应的视觉效果等级随机划分为训练集和测试集；以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型。

所述以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型，之后还包括：采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型。

所述采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型，之后还包括：以所述测试集内的压缩后的电脑屏幕图像为输入，以所述测试集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，验证所述修正后的深度学习网络模型。

本发明通过视觉信息保真度(Visual Information Fidelity，VIF)指标对不同的屏幕与不同的水印类型及水印强度进行水印效果等级划分，图片与水印效果等级作为先验信息，以训练深度学习网络模型。其中，视觉信息保真度是衡量测试样本与参考样本的差异性的指标，该指标广泛应用于图像增强、图像融合拼接、视频质量评估等领域，是流媒体巨头Netflix使用的视频流媒体质量评估工具VMAF的核心指标。

步骤104：对当前电脑屏幕添加不同水印强度的水印类型，获取实际电脑屏幕图像。

步骤105：将所述实际电脑屏幕图像输入所述深度学习网络模型中，输出所述实际电脑屏幕图像的视觉效果等级。

步骤106：根据所述实际电脑屏幕图像的视觉效果等级筛选出最佳视觉效果等级，并将所述最佳视觉效果等级所对应的水印类型作为对所述当前电脑屏幕添加的最佳水印类型；以所述最佳水印类型对应的水印强度值对所述当前电脑屏幕添加数字水印。

在实际应用中，所述以所述最佳水印类型对应的水印强度值对所述当前电脑屏幕添加数字水印，具体为：选取所述最佳水印类型对应的所有水印强度值的中间值，并以该所有水印强度值的中间值对当前电脑屏幕添加数字水印。

图2为本发明所提供的基于深度学习的电脑屏幕最佳水印类型确定系统结构图，如图2所示，一种基于深度学习的电脑屏幕最佳水印类型确定系统，包括：

电脑屏幕图像数据集获取模块201，用于对不同型号的电脑屏幕添加不同水印强度的水印类型，获取电脑屏幕图像数据集；所述电脑屏幕图像数据集包括多张电脑屏幕图像。

视觉效果等级等级确定模块202，用于对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

所述视觉效果等级等级确定模块202，具体包括：视觉信息保真度计算单元，用于计算每张所述电脑屏幕图像的每一类水印类型不同水印强度的视觉信息保真度；水印强度等级确定单元，用于根据所述视觉信息保真度确定水印强度等级；视觉效果等级确定单元，用于根据所述水印强度等级对每张所述电脑屏幕图像的视觉效果进行等级划分，确定所述电脑屏幕图像的视觉效果等级。

深度学习网络模型建立模块203，用于以所述电脑屏幕图像为输入，以所述电脑屏幕图像的视觉效果等级为输出，建立深度学习网络模型。

所述深度学习网络模型建立模块203，具体包括：图像压缩单元，用于将所述电脑屏幕图像数据集内的电脑屏幕图像进行压缩，确定压缩后的电脑屏幕图像；划分单元，用于按照8:2的比例，将所述压缩后的电脑屏幕图像以及所述压缩后的电脑屏幕图像所对应的视觉效果等级随机划分为训练集和测试集；深度学习网络模型建立单元，用于以所述训练集内的压缩后的电脑屏幕图像为输入，依次经过所述残差网络中的核心模块、卷积层以及池化层，以所述训练集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，建立深度学习网络模型。

本发明还包括：修正单元，用于采用均方误差损失，将水印强度值作为损失函数正则项，修正所述深度学习网络模型，确定修正后的深度学习网络模型。

验证单元，用于以所述测试集内的压缩后的电脑屏幕图像为输入，以所述测试集内的压缩后的电脑屏幕图像所对应的视觉效果等级为输出，验证所述修正后的深度学习网络模型。

实际电脑屏幕图像获取模块204，用于对当前电脑屏幕添加不同水印强度的水印类型，获取实际电脑屏幕图像。

实际电脑屏幕图像的视觉效果等级输出模块205，用于将所述实际电脑屏幕图像输入所述深度学习网络模型中，输出所述实际电脑屏幕图像的视觉效果等级。

最佳水印类型确定模块206，用于根据所述实际电脑屏幕图像的视觉效果等级筛选出最佳视觉效果等级，并将所述最佳视觉效果等级所对应的水印类型作为对所述当前电脑屏幕添加的最佳水印类型；以所述最佳水印类型对应的水印强度值对所述当前电脑屏幕添加数字水印。

本发明通过深度学习网络模型确定最佳水印类型，不受人为判断的影响。

本说明书中各个实施例采用递进的方式描述，每个实施例重点说明的都是与其他实施例的不同之处，各个实施例之间相同相似部分互相参见即可。对于实施例公开的系统而言，由于其与实施例公开的方法相对应，所以描述的比较简单，相关之处参见方法部分说明即可。

本文中应用了具体个例对本发明的原理及实施方式进行了阐述，以上实施例的说明只是用于帮助理解本发明的方法及其核心思想；同时，对于本领域的一般技术人员，依据本发明的思想，在具体实施方式及应用范围上均会有改变之处。综上所述，本说明书内容不应理解为对本发明的限制。