首页 / 专利库 / 视听技术与设备 / 块匹配运动估计 / 基于双MIC的OpenCL并行帧率上转换方法

基于双MIC的OpenCL并行率上转换方法

阅读:562发布:2020-05-15

专利汇可以提供基于双MIC的OpenCL并行率上转换方法专利检索,专利查询,专利分析的服务。并且本 发明 提出了一种基于双MIC的OpenCL并行 帧 率上转换方法,在保证图像 质量 的前提下,有效缩短帧率上转换的运行时间,提高帧率上转换的运行效率。实施步骤为:主线程对OpenCL设备的MIC1和MIC2进行初始化;主线程对读入的视频进行编号;主线程定义 信号 量并初始化;主线程在主机上开辟内存并创建子线程1和子线程2;子线程1控制MIC1,执行 运动估计 算法 ,同时子线程2控制MIC2,执行 运动补偿 算法,实现 视频帧 率的上转换;主线程关闭子线程1和子线程2。本发明有效地提高了算法的运行效率,可用于视频帧率上转换领域。,下面是基于双MIC的OpenCL并行率上转换方法专利的具体信息内容。

1.一种基于双MIC的OpenCL并行率上转换方法,包括如下步骤:
(1)主线程对OpenCL设备的MIC1和MIC2进行初始化,实现主机端对MIC设备的控制;
(2)主线程对读入的视频进行编号:主线程读入N帧视频,并对运动估计算法中视频当前帧的图像编号为i,初始化i=1,同时对运动补偿算法中视频当前帧的图像编号为j,初始化j=1,其中,i的取值范围是[1,N],j的取值范围是[1,N];
(3)主线程定义信号量并初始化:主线程定义信号量1和信号量2,并将信号量1的值初始化为1,信号量2的值初始化为0;
(4)主线程在主机上开辟内存并创建子线程:主线程在主机上开辟主机内存cpu_mem1、主机内存cpu_mem2和主机内存cpu_mem3,同时创建子线程1和子线程2;
(5)子线程1控制MIC1,执行运动估计算法:
(5a)子线程1在MIC1上开辟内存mic1_mem1和内存mic1_mem2;
(5b)子线程1将第i帧和第i+1帧的图像数据传输到内存mic1_mem1;
(5c)MIC1计算运动估计算法中第i帧图像数据的运动矢量MVi,并将MVi存入内存mic1_mem2中;
(5d)子线程1将MVi从内存mic1_mem2传入主机内存cpu_mem1;
(5e)子线程1判断信号量1的值是否大于0,若是,将信号量1的值减1,同时将主机内存cpu_mem1中的MVi写入主机内存cpu_mem2中,将信号量2的值加1,并执行步骤(5g),否则,执行步骤(5f);
(5f)子线程1等待子线程2修改信号量1的值,直到修改完成,并执行步骤(5e);
(5g)令i=i+1,子线程1判断i≤N是否成立,若是,执行步骤(5b),否则,子线程1挂起;
(6)子线程2控制MIC2,执行运动补偿算法,实现视频帧率的上转换:
(6a)子线程2在MIC2上开辟内存mic2_mem1、内存mic2_mem2和内存mic2_mem3;
(6b)子线程2将第j帧和第j+1帧的图像数据传入内存mic2_mem1;
(6c)子线程2判断信号量2的值是否大于0,若是,将信号量2的值减1,同时将主机内存cpu_mem2中的MVi读到内存mic2_mem2,并执行步骤(6e),否则,执行步骤(6d);
(6d)子线程2等待子线程1修改信号量2的值,直到修改完成,并执行步骤(6c);
(6e)MIC2计算待插帧中每个像素点的运动补偿插值,并将待插帧的插值结果存放在内存mic2_mem3中;
(6f)子线程2将插值结果从内存mic2_mem3传入主机内存cpu_mem3,并将主机内存cpu_mem3中的插值结果写文件到硬盘中,同时将信号量1的值加1;
(6g)令j=j+1,子线程2判断j≤N是否成立,若是,执行步骤(6b),否则,子线程2挂起;
(7)主线程关闭子线程1和子线程2。
2.根据权利要求1所述的基于双MIC的OpenCL并行帧率上转换方法,其特征在于,步骤(4)中所述的创建子线程1和子线程2,采用Pthread函数pthread_create。
3.根据权利要求1所述的基于双MIC的OpenCL并行帧率上转换方法,其特征在于,步骤(5a)中所述的子线程1在MIC1上开辟内存mic1_mem1和内存mic1_mem2,其与步骤(6a)中所述的子线程2在MIC2上开辟内存mic2_mem1、内存mic2_mem2和内存mic2_mem3,均采用OpenCL函数clCreateBuffer。
4.根据权利要求1所述的基于双MIC的OpenCL并行帧率上转换方法,其特征在于,步骤(5b)中所述的子线程1将第i帧和第i+1帧的图像数据传输到内存mic1_mem1,其与步骤(6b)所述的子线程2将第j帧和第j+1帧的图像数据传入内存mic2_mem1,均采用OpenCL函数clEnqueueWriteBuffer。
5.根据权利要求1所述的基于双MIC的OpenCL并行帧率上转换方法,其特征在于,步骤(5c)中所述的MIC1计算运动估计算法中第i帧图像数据的运动矢量MVi,实现步骤如下:
(5c1)将第i帧的图像划分为M×N2个宏,其中M≥1,N2≥1;
(5c2)计算每个宏块的候选运动矢量集;
(5c3)根据SAD计算公式,得出矢量集中每个候选运动矢量的SAD值;
其中X代表第i帧当前宏块的宽,Y代表第i帧当前宏块的高,xmn代表第i帧中当前块内位置为(m,n)点的像素值,ymn为第i+1帧中匹配块位置为(m,n)点的像素值;
(5c4)选择SAD值最小的候选运动矢量作为该宏块的运动矢量;
(5c5)依次计算完第i帧图像所有宏块的运动矢量。
6.根据权利要求1所述的基于双MIC的OpenCL 并行帧率上转换方法,其特征在于,步骤(5d)中所述的子线程1将MVi从内存mic1_mem2传入主机内存cpu_mem1,其与步骤(6f)中所述的子线程2将插值结果从内存mic2_mem3传入主机内存cpu_mem3,均采用OpenCL 函数cEnqueueReadBuffer。

说明书全文

基于双MIC的OpenCL并行率上转换方法

技术领域

[0001] 本发明属于视频处理技术领域,涉及一种OpenCL并行帧率上转换方法,具体涉及一种基于双MIC的OpenCL并行帧率上转换方法,适用于视频帧率上转换等领域。

背景技术

[0002] 近年来,视频领域的新技术不断涌现,通过增加视频的分辨率和提高视频的帧率等技术手段,已带给人们更清晰、更具冲击的视觉感受。比如从原先的标清视频,到现在的高清视频,甚至超高清的4K视频。现如今4K视频源也越来越多,已经逐渐进入到人们的视野当中,预示着人们对于视频画面的清晰度要求也更高了。2016年11月李安导演的新作《比利·林恩的中场战事》,首次尝试了“120帧/4K/3D”(每秒播放120帧、4K分辨率、3D效果)的技术,以每秒120帧高帧率的播放方式开创了电影技术的新法则,引起了业界的广泛关注。
[0003] 视频帧率上转换技术(FRUC),作为一种视频后处理手段,通过在原有视频帧中插入中间帧的方式,将低帧率视频转换成高帧率视频。最早在20世纪80年代在业界就出现了帧率转换的技术,线性内插帧被采用来执行帧率上转换算法,其中包括帧复制和帧平均,随后发展越来越成熟。90年代中期提出了基于运动补偿的视频帧率上转换技术,这种技术对运动的物体首先进行运动估计算法,尽可能获得接近真实运动的矢量场。它以或者像素点为单位,不同的块得到的矢量很可能不一样,所以包含了运动物体的各个部分,得到的矢量是比较精确的。然后根据得到的运动矢量,进行运动补偿模块的计算,即根据原始的视频帧和上一步得到的运动矢量通过插值的方式获得待插帧。
[0004] 随着视频图像从高清到超高清的迈进,以及高帧率视频的出现,使得处理视频的图像大小和帧数都有了极大的增长。图像规模的增大,算法复杂度的提高,使得原先算法处理的时间大大增加,无法满足迅速处理甚至实时的要求。因此,研究如何加速帧率上转换算法已成为当今比较迫切的问题之一。
[0005] Intel在2012年发布了Intel Xeon Phi协处理器,它是一款基于集成众核架构MIC(Many Intergrated Cores)的至强融合产品。该协处理器集成了50个以上的计算核心,并且具备了512bit的向量处理单元(VPU)。Intel MIC众核架构的硬件既保留了CPU中的多级流线,同时又配备了众多的计算核心,每个计算核心又可以并发执行4个线程,这就保证了MIC具有同时处理多个任务的优势。2015年焦文在硕士论文中提出了在MIC架构上的并行运动估计和并行运动补偿方法,该方法利用OpenMP的fork-join模型,将运动估计的每个块求解运动矢量的过程放在MIC每个线程执行,运动补偿模块的每个像素点放在MIC每个线程执行。该方法虽然获得了一定的加速效果,但是,并没有研究在双MIC上如何实现视频帧率上转换。

发明内容

[0006] 本发明的目的在于针对上述已有技术的不足,提出了一种基于双MIC的OpenCL并行帧率上转换方法,在保证图像质量的前提下,有效缩短帧率上转换的运行时间,提高帧率上转换的运行效率。
[0007] 为实现上述目的,本发明采取的技术方案包括有如下步骤:
[0008] (1)主线程对OpenCL设备的MIC1和MIC2进行初始化,实现主机端对MIC设备的控制;
[0009] (2)主线程对读入的视频进行编号:主线程读入N帧视频,并对运动估计算法中视频当前帧的图像编号为i,初始化i=1,同时对运动补偿算法中视频当前帧的图像编号为j,初始化j=1,其中,i的取值范围是[1,N],j的取值范围是[1,N];
[0010] (3)主线程定义信号量并初始化:主线程定义信号量1和信号量2,并将信号量1的值初始化为1,信号量2的值初始化为0;
[0011] (4)主线程在主机上开辟内存并创建子线程:主线程在主机上开辟主机内存cpu_mem1、主机内存cpu_mem2和主机内存cpu_mem3,同时创建子线程1和子线程2;
[0012] (5)子线程1控制MIC1,执行运动估计算法:
[0013] (5a)子线程1在MIC1上开辟内存mic1_mem1和内存mic1_mem2;
[0014] (5b)子线程1将第i帧和第i+1帧的图像数据传输到内存mic1_mem1;
[0015] (5c)MIC1计算运动估计算法中第i帧图像数据的运动矢量MVi,并将MVi存入内存mic1_mem2中;
[0016] (5d)子线程1将MVi从内存mic1_mem2传入主机内存cpu_mem1;
[0017] (5e)子线程1判断信号量1的值是否大于0,若是,将信号量1的值减1,同时将主机内存cpu_mem1中的MVi写入主机内存cpu_mem2中,将信号量2的值加1,并执行步骤(5g),否则,执行步骤(5f);
[0018] (5f)子线程1等待子线程2修改信号量1的值,直到修改完成,并执行步骤(5e);
[0019] (5g)令i=i+1,子线程1判断i≤N是否成立,若是,执行步骤(5b),否则,子线程1挂起;
[0020] (6)子线程2控制MIC2,执行运动补偿算法,实现视频帧率的上转换:
[0021] (6a)子线程2在MIC2上开辟内存mic2_mem1、内存mic2_mem2和内存mic2_mem3;
[0022] (6b)子线程2将第j帧和第j+1帧的图像数据传入内存mic2_mem1;
[0023] (6c)子线程2判断信号量2的值是否大于0,若是,将信号量2的值减1,同时将主机内存cpu_mem2中的MVi读到内存mic2_mem2,并执行步骤(6e),否则,执行步骤(6d);
[0024] (6d)子线程2等待子线程1修改信号量2的值,直到修改完成,并执行步骤(6c);
[0025] (6e)MIC2计算待插帧中每个像素点的运动补偿插值,并将待插帧的插值结果存放在内存mic2_mem3中;
[0026] (6f)子线程2将插值结果从内存mic2_mem3传入主机内存cpu_mem3,并将主机内存cpu_mem3中的插值结果写文件到硬盘中,同时将信号量1的值加1;
[0027] (6g)令j=j+1,子线程2判断j≤N是否成立,若是,执行步骤(6b),否则,子线程2挂起;
[0028] (7)主线程关闭子线程1和子线程2。
[0029] 本发明与现有技术相比,具有如下优点:
[0030] 1、本发明通过Pthread在主机端创建子线程1和子线程2,子线程1控制MIC1计算当前帧图像的运动矢量,同时子线程2控制MIC2计算当前帧的上一帧图像的运动补偿插值,避免了现有技术通过单个计算设备串行计算运动矢量和运动补偿插值耗时大的缺陷,有效地提高了帧率上转换的运行效率。
[0031] 2、本发明中子线程1和子线程2都需要访问存放运动矢量的主机内存,其中子线程1利用信号量控制对这块主机内存进行写操作,子线程2利用信号量控制对这块主机内存进行读操作,避免了两个线程对这块内存的读写冲突,保证了帧率上转换的正确性。
附图说明
[0032] 图1是本发明的实现流程图
[0033] 图2是本发明对OpenCL设备初始化的实现流程图;
[0034] 图3是本发明子线程1控制MIC1执行运动估计算法的实现流程图;
[0035] 图4是本发明子线程2控制MIC2执行运动补偿算法的实现流程图;
[0036] 图5是本发明仿真实验输入的不同分辨率的单帧测试视频图;
[0037] 图6是本发明判别正确性的仿真实验结果图。

具体实施方式

[0038] 下面结合附图和具体实施例,对本发明作进一步说明。
[0039] 参照图1,基于双MIC的OpenCL并行帧率上转换方法,包括如下步骤:
[0040] 步骤1)主线程对OpenCL设备的MIC1和MIC2进行初始化,实现主机端对MIC设备的控制:程序开始执行后,进入主线程。对OpenCL设备初始化的实现流程图参照图2,主线程在获取执行OpenCL设备阶段时,通过device[0]来获取MIC1的设备信息,通过device[1]来获取MIC2的设备信息。然后对MIC1设备创建命令队列commandqueue1,对MIC2设备创建命令队列commandqueue2,在运动估计算法中,利用commandqueue1控制对MIC1的数据读写和执行运动估计kernel的操作。在运动补偿算法中,利用commandqueue2控制对MIC2的数据读写和执行运动补偿kernel的操作。
[0041] 步骤2)主线程对读入的视频进行编号:主线程读入30帧视频,由于执行运动估计算法的子线程1和执行运动补偿算法的子线程2,并不是对同一帧图像并行处理,所以对运动估计算法中视频当前帧的图像编号为i,初始化i=1,同时对运动补偿算法中视频当前帧的图像编号为j,初始化j=1,其中,i的取值范围是[1,30],j的取值范围是[1,30]。
[0042] 步骤3)主线程定义信号量并初始化:为了防止两个线程对同一块内存的访问冲突,本发明采用信号量控制两个线程之间的通信,主线程定义信号量1和信号量2,将信号量1的值初始化为1,信号量2的值初始化为0。
[0043] 步骤4)主线程在主机上开辟内存并创建子线程:主线程在主机上开辟主机内存cpu_mem1、主机内存cpu_mem2和主机内存cpu_mem3,采用Pthread函数pthread_create创建子线程1和子线程2。
[0044] 步骤5)子线程1控制MIC1,执行运动估计算法,实现流程图参照图3:
[0045] 步骤5a)子线程1采用OpenCL函数clCreateBuffer在MIC1上开辟内存mic1_mem1和内存mic1_mem2,mic1_mem1用来存储输入的图像数据,mic1_mem2用来存储计算出的运动矢量结果。
[0046] 步骤5b)子线程1采用OpenCL函数clEnqueueWriteBuffer将第i帧和第i+1帧的图像数据传输到内存mic1_mem1。
[0047] 步骤5c)MIC1计算运动估计算法中第i帧图像数据的运动矢量MVi,并将MVi存入内存mic1_mem2中:将第i帧的图像划分为240×135个宏块,计算每个宏块的候选运动矢量集,根据SAD计算公式,得出矢量集中每个候选运动矢量的SAD值,选择SAD值最小的候选运动矢量作为该宏块的运动矢量,依次计算完第i帧图像所有宏块的运动矢量。
[0048]
[0049] 其中X代表第i帧当前宏块的宽,Y代表第i帧当前宏块的高,xmn代表第i帧中当前块内位置为(m,n)点的像素值,ymn为第i+1帧中匹配块位置为(m,n)点的像素值。
[0050] 步骤5d)子线程1采用OpenCL函数clEnqueueReadBuffer将MVi从内存mic1_mem2传入主机内存cpu_mem1。
[0051] 步骤5e)子线程1判断信号量1的值是否大于0,若是,表明子线程2已经计算完第j帧的运动补偿,可以给子线程2中存储输入数据的内存块拷贝数据,将信号量1的值减1,同时将主机内存cpu_mem1中的MVi写入主机内存cpu_mem2中,将信号量2的值加1,并执行步骤5g),否则,执行步骤5f)。
[0052] 步骤5f)子线程1等待子线程2修改信号量1的值,直到修改完成,并执行步骤5e)。
[0053] 步骤5g)令i=i+1,子线程1判断i≤30是否成立,若是,计算下一帧图像的运动矢量,执行步骤5b),否则,表明视频所有帧的运动矢量已经计算完成,子线程1挂起。
[0054] 步骤6)子线程2控制MIC2,执行运动补偿算法,实现流程图参照图4:
[0055] 步骤6a)子线程2采用OpenCL函数clCreateBuffer在MIC2上开辟内存mic2_mem1、内存mic2_mem2和内存mic2_mem3。
[0056] 步骤6b)子线程2采用OpenCL函数clEnqueueWriteBuffer将第j帧和第j+1帧的图像数据传入内存mic2_mem1。
[0057] 步骤6c)子线程2判断信号量2的值是否大于0,若是,表明子线程1已经将MVi从主机内存cpu_mem1写入主机内存cpu_mem2,将信号量2的值减1,同时将主机内存cpu_mem2中的MVi读到内存mic2_mem2,并执行步骤6e),否则,执行步骤6d)。
[0058] 步骤6d)子线程2等待子线程1修改信号量2的值,直到修改完成,并执行步骤6c)。
[0059] 步骤6e)MIC2上计算待插帧中每个像素点的插值结果,并将待插帧的插值结果存放在内存mic2_mem3中。
[0060] 步骤6f)子线程2采用OpenCL函数clEnqueueReadBuffer将插值结果从内存mic2_mem3传入主机内存cpu_mem3,并将主机内存cpu_mem3中的插值结果写文件到硬盘中,同时将信号量1的值加1,表明第j帧的运动补偿插值计算完成。
[0061] 步骤6g)令j=j+1,子线程2判断j≤N是否成立,若是,计算下一帧图像的运动补偿插值,执行步骤(6b),否则,表明视频所有的待插帧已经计算完成,子线程2挂起。
[0062] 步骤7)主线程关闭子线程1和子线程2。
[0063] 以下结合仿真实验,对本发明的技术效果进一步说明:
[0064] 1)仿真条件:
[0065] 仿真实验输入的不同分辨率的单帧测试视频图,参照图5,图5(a)是2K分辨率的单帧测试视频图ParkScene_1920×1080,图5(b)是4K分辨率的单帧测试视频图Sunset_3840×2160。
[0066] 仿真实验环境使用西安电子科技大学高性能计算中心集群设备,测试平台参数见表1所示。
[0067] 表1
[0068]
[0069] 2)仿真内容与结果分析:
[0070] 基于双MIC的OpenCL并行帧率上转换方法,表2为串行算法和本发明并行算法的PSNR值对比,表3为运动估计算法和运动补偿算法测试时间,表4为串行测试时间和本发明测试时间。
[0071] 表2
[0072]视频序列 串行算法 本发明并行算法
2K视频 36.39 36.37
4K视频 38.43 38.40
[0073] 表3
[0074]测试序列 运动估计算法(ms) 运动补偿算法(ms)
2K视频 240.52 187.92
4K视频 249.37 684.38
[0075] 表4
[0076]测试序列 串行算法(ms) 本发明并行方法(ms)
2K视频 428.44 261.30
4K视频 933.75 689.61
[0077] 图6为判别正确性的仿真实验结果图,图6(a)是2K分辨率视频的单帧仿真结果图,图6(b)是4K分辨率视频的单帧仿真结果图。为了更加准确判断本发明的并行方案的正确性,采用客观评价准则的峰值信噪比(PSNR)用来评价算法的质量,本发明仿真结果的PSNR值如表2所示,在本发明中提出的并行帧率上转换算法得到的PSNR和串行算法的PSNR接近,可以得到结论,并行帧率上转换算法的图像质量与串行算法的图像质量保持一致,因此验证了本次并行帧率上转换算法的正确性。
[0078] 表3为帧率上转换算法中运动估计算法和运动补偿算法的串行测试时间。
[0079] 表4为帧率上转换算法的串行测试时间和本发明的并行测试时间,其中串行测试时间为表3中运动估计算法时间和运动补偿算法时间的求和结果,本发明的测试时间约为表3中运动估计算法测试时间和运动补偿算法测试时间两者之间的较大者,再加上数据拷贝时间和信号量等待时间。从表4可以看出,本发明的测试时间有效地加速了帧率上转换算法的计算效率。
高效检索全球专利

专利汇是专利免费检索,专利查询,专利分析-国家发明专利查询检索分析平台,是提供专利分析,专利查询,专利检索等数据服务功能的知识产权数据服务商。

我们的产品包含105个国家的1.26亿组数据,免费查、免费专利分析。

申请试用

分析报告

专利汇分析报告产品可以对行业情报数据进行梳理分析,涉及维度包括行业专利基本状况分析、地域分析、技术分析、发明人分析、申请人分析、专利权人分析、失效分析、核心专利分析、法律分析、研发重点分析、企业专利处境分析、技术处境分析、专利寿命分析、企业定位分析、引证分析等超过60个分析角度,系统通过AI智能系统对图表进行解读,只需1分钟,一键生成行业专利分析报告。

申请试用

QQ群二维码
意见反馈