[發明專利]基于GPU的視頻隱形水印嵌入方法及系統

在審

• 申請號: 202210322703.0
• 申請日: 2022-03-30
• 公開（公告）號: CN114598938A
• 公開（公告）日: 2022-06-07
• 發明（設計）人: 田輝;劉海飛;郭玉剛;張志翔
• 申請（專利權）人: 合肥高維數據技術有限公司
• 主分類號: H04N21/8358
• 分類號: H04N21/8358;H04N19/467
• 代理公司: 暫無信息
• 代理人: 暫無信息

1.一種基于GPU的視頻隱形水印嵌入方法，其特征在于：包括如下步驟：

S100、根據GPU型號確定其處理數據的最佳格式；

S200、將視頻數據使用GPU解碼成該最佳格式得到多幀視頻圖像；

S300、將隱形水印數據轉換成該最佳格式并與每一幀視頻圖像進行疊加；

S400、將添加水印后的多幀視頻圖像在GPU的編碼芯片中進行編碼得到含有隱形水印的視頻數據。

2.如權利要求1所述的基于GPU的視頻隱形水印嵌入方法，其特征在于：所述的最佳格式為I420、YV12、NV12、NV21中的任一種。

3.如權利要求1所述的基于GPU的視頻隱形水印嵌入方法，其特征在于：所述的步驟S200中，視頻數據通過如下方式獲得：

S210、讀取待嵌入水印的視頻文件的后綴名，判斷文件的封裝格式；

S220、根據文件的封裝格式，從視頻文件中分離得到音頻數據和視頻數據，并將音頻數據存儲在臨時文件中；

步驟S400后，還包括如下步驟對得到的視頻數據進行處理：

S500、將臨時文件中存儲的音頻數據和嵌入水印的視頻數據進行重新封裝得到嵌入水印的視頻文件。

4.如權利要求3所述的基于GPU的視頻隱形水印嵌入方法，其特征在于：所述的步驟S100中，最佳格式為NV12。

5.如權利要求4所述的基于GPU的視頻隱形水印嵌入方法，其特征在于：所述的步驟S220中，包括如下步驟：

S221、創建H264/H265解碼器；

S222、將H264/H265數據拷貝到顯存中；

S223、將顯存中的數據傳輸到GPU內置的解碼器函數中，從而獲取到解碼器解碼過后的NV12數據；

所述的步驟S300中，將顯存中通過水印產生的NV12數據與步驟S223產生的NV12數據按照邏輯關系進行數據疊加；

所述的步驟S400中，將上述疊加水印后的NV12數據輸入到GPU專用的編碼器中，編碼器輸出的數據就是壓縮過后的H264/H265視頻數據。

6.如權利要求3所述的基于GPU的視頻隱形水印嵌入方法，其特征在于：步驟S220中，若文件的封裝格式為mov，則通過FFmpeg CPU解碼器進行解碼，否則，直接利用GPU進行解碼。

7.一種基于GPU的視頻隱形水印嵌入系統，其特征在于：包括設置在GPU中的圖片轉換單元、圖片存儲單元、視頻解碼單元、數據疊加單元以及視頻編碼單元；所述的圖片轉換單元用于在GPU顯存中將水印矩陣圖轉換成NV12格式，圖片存儲單元用于存儲轉換后的圖片數據，視頻解碼單元用于在GPU編解碼芯片中將原視頻解碼為NV12格式，數據疊加單元用于在GPU顯存中將水印和視頻的NV12數據進行疊加操作，視頻編碼單元用于對修改后NV12數據流在GPU編解碼芯片中編碼得到視頻信息并輸出。

8.一種計算機可讀存儲介質，其特征在于：其上存儲有計算機程序，所述計算機程序被處理器執行時，實現如權利要求1-6中任一項所述的基于GPU的視頻隱形水印嵌入方法。

9.一種電子設備，其特征在于：包括存儲器、處理器及存儲在存儲器上的計算機程序，所述處理器執行所述計算機程序時，實現如權利要求1-6中任一項所述的基于GPU的視頻隱形水印嵌入方法。