[發明專利]數字視頻水印方法和裝置、設備及存儲介質

在審

• 申請號: 202310305125.4
• 申請日: 2023-03-24
• 公開（公告）號: CN116471362A
• 公開（公告）日: 2023-07-21
• 發明（設計）人: 危學濤;葉冠輝
• 申請（專利權）人: 南方科技大學
• 主分類號: H04N1/32
• 分類號: H04N1/32;H04N21/8358
• 代理公司: 廣州嘉權專利商標事務所有限公司 44205
• 代理人: 洪銘福

1.一種數字視頻水印方法，其特征在于，所述方法包括：

獲取原始視頻數據和原始水印信息；其中，所述原始視頻數據包括：通道維度信息和時間維度信息；

將所述原始視頻數據的所述通道維度信息和所述時間維度信息進行合并處理，得到訓練視頻數據；

將所述訓練視頻數據和所述原始水印信息輸入至預設的原始數字視頻水印網絡進行編碼處理，得到原始水印視頻數據；其中，所述原始數字視頻水印網絡通過對數字圖片神經網絡的數據通道數修改得到；

將預設的噪聲數據增加至所述原始水印視頻數據，得到噪聲水印視頻數據；其中，所述噪聲數據從預設的噪聲數據庫隨機選取；

將所述噪聲水印視頻數據輸入至所述原始數字視頻水印網絡進行解碼處理，得到預測水印信息；

根據所述訓練視頻數據、所述原始水印視頻數據、所述預測水印信息和所述原始水印信息進行損失計算，得到目標損失數據；

根據所述目標損失數據對所述原始數字視頻水印網絡進行參數調整，得到目標數字視頻水印網絡；

獲取目標視頻數據，將所述目標視頻數據輸入至所述目標數字視頻水印網絡進行水印植入，得到目標水印視頻數據。

2.根據權利要求1所述的方法，其特征在于，在所述獲取目標視頻數據，將所述目標視頻數據輸入至所述目標數字視頻水印網絡進行水印植入，得到目標水印視頻數據之后，所述方法還包括：

獲取待處理水印視頻數據；

將所述待處理水印視頻數據輸入至所述目標數字視頻水印網絡進行解碼處理，得到目標水印信息。

3.根據權利要求1所述的方法，其特征在于，所述原始數字視頻水印網絡包括：編碼器和解碼器；

所述將所述訓練視頻數據和所述原始水印信息輸入至預設的原始數字視頻水印網絡進行編碼處理，得到原始水印視頻數據，包括：

將所述訓練視頻數據和所述原始水印信息輸入至所述編碼器進行編碼處理，得到所述原始水印視頻數據；

所述將所述噪聲水印視頻數據輸入至所述原始數字視頻水印網絡進行解碼處理，得到預測水印信息，包括：

將所述噪聲水印視頻數據輸入至所述解碼器進行解碼處理，得到所述預測水印信息。

4.根據權利要求1所述的方法，其特征在于，所述將預設的噪聲數據增加至所述原始水印視頻數據，得到噪聲水印視頻數據，包括：

獲取所述噪聲數據的噪聲信息；

若所述噪聲信息表征為非H264噪聲，將所述噪聲數據加入所述原始水印視頻數據，得到所述噪聲水印視頻數據；

若所述噪聲信息表征為H264噪聲，根據預設的前向攻擊模擬層和所述噪聲數據，或者根據所述訓練視頻數據對所述原始水印視頻數據進行噪聲加入，得到所述噪聲水印視頻數據。

5.根據權利要求4所述的方法，其特征在于，所述若所述噪聲信息表征為H264噪聲，根據預設的前向攻擊模擬層和所述噪聲數據，或者根據所述訓練視頻數據對所述原始水印視頻數據進行噪聲加入，得到所述噪聲水印視頻數據，包括：

若所述噪聲信息表征為H264噪聲，根據所述訓練視頻數據進行噪聲模擬，得到模擬噪聲數據，并將所述模擬噪聲數據輸入至所述原始水印視頻數據，得到所述噪聲水印視頻數據；

或者，

若所述噪聲信息表征為H264噪聲，將所述噪聲數據輸入至所述原始水印視頻數據得到噪聲水印視頻數據，根據所述前向攻擊模擬層對所述原始水印視頻數據和所述訓練視頻數據進行差值計算，得到視頻數據差值，將所述視頻數據差值設置為不參與所述原始數字視頻水印網絡訓練過程梯度的反向傳播，并將所述視頻數據差值加入至所述原始水印視頻數據，得到所述噪聲水印視頻數據。