[发明专利]图像中水印的检测方法、装置、电子设备及存储介质

权利要求书

1.一种图像中水印的检测方法，其特征在于，包括：

采用预设的特征提取层，提取待检测图像的特征图；

根据预设锚框和区域推荐网络，在所述特征图中确定预设数量的候选检测区域，其中，所述预设锚框根据第一样本图像训练获取，所述第一样本图像中包含标记的水印区域；

根据水印检测算法和所述预设数量的候选检测区域，确定所述待检测图像中是否包括水印区域。

2.根据权利要求1所述的方法，其特征在于，所述根据水印检测算法和所述预设数量的候选检测区域，确定所述待检测图像中是否包括水印区域，包括：

根据水印检测算法，检测所述预设数量的候选检测区域中是否包括水印区域；

若所述候选检测区域中包括水印区域，则获取所述水印区域的位置。

3.根据权利要求1或2所述的方法，其特征在于，所述根据水印检测算法和所述预设数量的候选检测区域，确定所述待检测图像中是否包括水印区域之后，所述方法还包括：

若所述待检测图像中包括水印区域，采用预设水印检测模型，确定所述待检测图像中所述水印的类别，其中，所述水印检测模型由第二样本图像训练获取，所述第二样本图像包含水印区域且标记有水印的类别。

4.根据权利要求1所述的方法，其特征在于，所述根据水印检测算法和所述预设数量的候选检测区域，确定所述待检测图像中是否包括水印区域，包括：

采用预设的特征图尺寸调节算法对所述预设数量的候选检测区域的特征图进行处理，获取预设数量、相同尺寸的待检测区域的特征图；

根据所述水印检测算法和所述预设数量的待检测区域的特征图，确定所述待检测图像中是否包括水印区域。

5.根据权利要求3所述的方法，其特征在于，所述待检测图像包括水印区域，所述根据水印检测算法和所述预设数量的待检测区域的特征图，确定所述待检测图像中的水印区域，包括：

根据预设的非极大值抑制算法和预设数量、相同尺寸的所述待检测区域的特征图，确定所述待检测图像中的水印区域。

6.根据权利要求1所述的方法，其特征在于，包括：

获取第一训练样本，所述第一训练样本包括：第一样本图像，所述第一样本图像中包含标记的水印区域；

根据预设的第一全连接网络和预设的回归损失函数，训练获取所述水印检测算法。

7.根据权利要求3所述的方法，其特征在于，包括：

获取第二训练样本，所述第二训练样本包括：第二样本图像，所述第二样本图像包含水印区域且标记有水印的类别；

根据预设的第二全连接网络和预设的分类损失函数，训练获取所述水印检测模型。

8.根据权利要求1或5所述的方法，其特征在于，所述方法还包括：

获取所述第一样本图像中包含标记的水印区域的形状和尺寸；

根据所述第一样本图像中包含标记的水印区域的形状和尺寸，确定所述预设锚框的形状和尺寸。

9.一种图像中水印的检测装置，其特征在于，包括：提取模块、第一确定模块及第二确定模块；

所述提取模块，用于采用预设的特征提取层，提取待检测图像的特征图；

所述第一确定模块，用于根据预设锚框和区域推荐网络，在所述特征图中确定预设数量的候选检测区域，其中，所述预设锚框根据第一样本图像训练获取，所述第一样本图像中包含标记的水印区域；

所述第二确定模块，用于根据水印检测算法和所述预设数量的候选检测区域，确定所述待检测图像中是否包括水印区域。

10.一种电子设备，其特征在于，包括：处理器、存储介质和总线，所述存储介质存储有所述处理器可执行的机器可读指令，当电子设备运行时，所述处理器与所述存储介质之间通过总线通信，所述处理器执行所述机器可读指令，以执行如权利要求1-8任一所述图像中水印的检测方法的步骤。

11.一种存储介质，其特征在于，所述存储介质上存储有计算机程序，所述计算机程序被处理器运行时执行如权利要求1-8任一所述图像中水印的检测方法的步骤。