[發明專利]基于水印數據增強的圖片隱形水印檢測方法

在審

• 申請號: 202211087893.9
• 申請日: 2022-09-07
• 公開（公告）號: CN115393647A
• 公開（公告）日: 2022-11-25
• 發明（設計）人: 田輝;劉其開;郭玉剛;張志翔
• 申請（專利權）人: 合肥高維數據技術有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/80;G06V10/82;G06V10/774;G06N3/04;G06N3/08;G06T1/00
• 代理公司: 暫無信息
• 代理人: 暫無信息

1.一種基于水印數據增強的圖片隱形水印檢測方法，其特征在于：包括如下步驟：

獲得訓練好的水印增強網絡和分類網絡；

將待檢測圖片輸入訓練好的水印增強網絡中得到水印概率圖；

將待檢測圖片和水印概率圖進行通道融合，并將融合后的四通道圖輸入至訓練好的分類網絡中得到分類結果，分類結果為有水印或無水印。

2.如權利要求1所述的基于水印數據增強的圖片隱形水印檢測方法，其特征在于：所述的獲得訓練好的水印增強網絡和分類網絡包括：

構建水印增強網絡和分類網絡；

分別準備第一樣本數據集和第二樣本數據集；

分別確定兩個網絡的損失函數；

利用第一樣本數據集訓練水印增強網絡得到訓練好的水印增強網絡，利用第二樣本數據集訓練分類網絡得到訓練好的分類網絡。

3.如權利要求2所述的基于水印數據增強的圖片隱形水印檢測方法，其特征在于：所述的分別準備第一樣本數據集和第二樣本數據集包括如下步驟：

準備無水印的圖片作為負樣本圖片，負樣本圖片對應的水印概率圖中任一像素均標記為0；

根據水印生成算法，對無水印的圖片進行水印嵌入操作，得到帶隱形水印的正樣本圖片，并根據生成算法直接生成正樣本圖片的水印概率圖，正樣本圖片對應的水印概率圖中含水印的像素區域標記為1、無水印的像素區域標記為0；

將正樣本圖片、負樣本圖片分別與其對應的水印概率圖進行通道融合得到四通道圖；

正樣本圖片及其對應的水印概率圖構成第一樣本數據集；四通道圖及其對應的類別標簽構成第二樣本數據集。

4.如權利要求3所述的基于水印數據增強的圖片隱形水印檢測方法，其特征在于：對所述的正樣本圖片和負樣本圖片按如下任一種方法進行處理：圖像剪裁拼接處理、邊緣鏡像填充處理、壓縮處理、高斯噪音以及高斯模糊濾波處理。

5.如權利要求1所述的基于水印數據增強的圖片隱形水印檢測方法，其特征在于：所述的水印增強網絡以Unet網絡為基礎，并按如下步驟進行改進：

將Unet網絡中的max pool操作替換為空洞抽樣池化加第一1*1的卷積層，空洞抽樣池化將C1×H1×W1大小的特征圖處理成(4×C1)×(H1/2)×(W1/2)大小，第一1*1卷積層將(4×C1)×(H1/2)×(W1/2)大小的特征圖降低為C1×(H1/2)×(W1/2)大小；

將Unet網絡中的上采樣操作替換為第二1*1卷積層加空洞上采樣操作，第二1*1卷積層用于擴充特征圖的通道數至原通道數的4倍，空洞上采樣操作為空洞抽樣池化操作的逆處理。

6.一種計算機可讀存儲介質，其特征在于：其上存儲有計算機程序，所述計算機程序被處理器執行時，實現如權利要求1-5中任一項所述的基于水印數據增強的圖片隱形水印檢測方法。

7.一種電子設備，其特征在于：包括存儲器、處理器及存儲在存儲器上的計算機程序，所述處理器執行所述計算機程序時，實現如權利要求1-5中任一項所述的基于水印數據增強的圖片隱形水印檢測方法。