[发明专利]预测单元尺寸决策的方法、装置、电子设备及介质

说明书

本申请公开了一种预测单元尺寸决策的方法、装置、电子设备及介质。本申请中，可以基于当前编码单元的运动估计结果以及预测单元决策模型，得到多个第一候选尺寸预测单元；基于整像素运动估计过程确定的预测单元决策结果，在多个第一候选尺寸预测单元中确定进入分像素运动估计阶段的第二候选尺寸预测单元；根据分像素运动估计阶段中的运动信息，从第二候选尺寸预测单元中确定目标预测单元。通过应用本申请的技术方案，可以在IME阶段提前对帧间CTU中每个CU所有可能的PU划分模式进行判断，并将划分结果传递给FME和MD，能显著减少进入两者的候选划分数量，减少所需的存储代价以及FME的电路面积。

技术领域

本申请中涉及图像编解码处理技术，尤其是一种预测单元尺寸决策的方法、 装置、电子设备及介质。

背景技术

针对目前主流的编码标准通常是采用CTU结构。一种方式中，每个CTU支 持编码单元CU的尺寸为64x64到8x8。在标准实现中，无论是帧内编码还是帧 间编码，最终当前编码单元CU在模式决策过程中，通常需要经过大量的率失真 优化计算处理和决策后才能决定当前CU划分的尺寸模式。

然而，通常这个过程是整个编码器最复杂的处理过程，需要占据整个编码 器一半以上的处理时间。所以，为了实现能够高效的软件和硬件编码器产品， 需要研究高效的快速PU尺寸提前决策方法。

发明内容

本申请实施例提供一种预测单元尺寸决策的方法、装置、电子设备及介质， 其中，根据本申请实施例的一个方面，提供的一种预测单元尺寸决策的方法， 其特征在于，包括：

基于当前编码单元的运动估计结果以及预测单元决策模型，得到多个第一 候选尺寸预测单元；

基于整像素运动估计过程确定的预测单元决策结果，在所述多个第一候选 尺寸预测单元中确定进入分像素运动估计阶段的第二候选尺寸预测单元；

根据所述分像素运动估计阶段中的运动信息，从所述第二候选尺寸预测单 元中确定目标预测单元。 可选地，在基于本申请上述方法的另一个实施例中，在所述基于当前编码 单元的运动估计结果以及预测单元决策模型，得到多个第一候选尺寸预测单元 之前，还包括： 利用哈达玛变换差距绝对值和技术，分别计算当前编码单元的运动估计代价； 确定每个当前编码单元对应的第一运动矢量、第一哈达玛域残差系数信息 以及像素差距代价； 基于所述每个当前编码单元对应的第一运动矢量、第一哈达玛域残差系数 信息以及像素差距代价，建立所述预测单元决策模型。 可选地，在基于本申请上述方法的另一个实施例中，利用如下公式分别计 算当前编码单元的运动估计代价：

J_motion＝SATD+λ_motion·MVBits

其中，SATD代表当前编码单元与其对应的预测单元之间的哈达玛变换后的 代价，λ_motion为运动估计拉格朗日乘子，MVBits表示运动矢量对应 的比特数。

可选地，在基于本申请上述方法的另一个实施例中，利用如下公式建立所 述预测单元决策模型：

J_RDO＝D+λ_mode·(MVBits+ModeBits+R)

其中，ModeBits表示预测单元划分模式对应的估计比特数，R代表当 前预测单元对应的残差系数产生的比特数，λ_mode为模式决策拉格朗日乘子。

可选地，在基于本申请上述方法的另一个实施例中，所述根据所述分像素 运动估计阶段中的运动信息，从所述第二候选尺寸预测单元中确定目标预测单 元，包括：

比较当前编码单元内的各个第二候选尺寸预测单元的划分方式，选择其中 运动估计代价低于预设阈值的第二候选尺寸预测单元作为所述目标预测单元。