[发明专利]基于GPU的视频隐形水印嵌入方法及系统

权利要求书

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的视频隐形水印嵌入方法。