[发明专利]一种预测模式的选择方法及装置、可读存储介质

说明书

本申请提供一种预测模式的选择方法及装置、可读存储介质。该方法包括：确定待处理的图像帧PU单元的多种PU预测模式；基于所述图像帧的亮度信息计算所述多种PU预测模式分别对应的亮度率失真代价；根据所述多种PU预测模式分别对应的亮度率失真代价筛选出需要计算色度率失真代价的目标预测模式，并基于所述图像帧的色度信息计算所述目标预测模式对应的色度率失真代价；根据所述多种PU预测模式分别对应的亮度率失真代价和所述目标预测模式对应的色度率失真代价确定所述图像帧PU单元的最优预测模式。该方法实现了预测流程的优化，降低了模式选择的复杂度，适用性强。

技术领域

本申请涉及视频编码技术领域，具体而言，涉及一种预测模式的选择方法及装置、可读存储介质。

背景技术

在视频编码中，在进行帧预测时，需要基于多个预测模式(划分方式)选择一个预测模式，然后根据选择的预测模式进行编码。选择预测模式的过程涉及到各种计算，目前常见的预测模式选择的方案有三种：一是不进行优化，所有的预测模式都进行计算；二是优先进行对称划分模式的计算，然后根据一定的优化策略，选择性地跳过部分非对称划分方式。第三种为根据一些统计特性以及阈值判断，策略性选择某几种预测模式进行计算和判断。

对于第一种方式，虽然能够保证编码质量，但是编码复杂度太高，编码时间太长，编码需要的资源量高。对于第二种和第三种，只是简单的利用数值进行比较和判断，并且舍弃了大多数划分模式，最后得到的预测模式的适用性较差。

可见，现有技术的模式选择的方式复杂度高，适用性较差。

发明内容

本申请实施例的目的在于提供一种预测模式的选择方法及装置、可读存储介质，用以提高预测模式选择的适用性和降低计算复杂度。

第一方面，本申请实施例提供一种预测模式的选择方法，包括：确定待处理的图像帧PU单元的多种PU预测模式；基于所述图像帧的亮度信息计算所述多种PU预测模式分别对应的亮度率失真代价；根据所述多种PU预测模式分别对应的亮度率失真代价筛选出需要计算色度率失真代价的目标预测模式，并基于所述图像帧的色度信息计算所述目标预测模式对应的色度率失真代价；根据所述多种PU预测模式分别对应的亮度率失真代价和所述目标预测模式对应的色度率失真代价确定所述图像帧PU单元的最优预测模式。

在本申请实施例中，在确定图像帧的多种PU预测模式后，先基于亮度信息计算亮度率失真代价，然后筛选出一个目标预测模式；对于色度，只计算在该目标预测模式下的色度率失真代价，最后再根据计算出的各个率失真代价确定图像帧的预测模式。与现有技术相比，考虑到人眼视觉体验，色度信号损失相对不敏感，色度信息的计算利用亮度信息计算的结果进行计算，不但减少色度信息的计算量，对最终的结果的影响可以忽略不计。对亮度信息与色度信息采取不同比重的计算量，即亮度进行全划分全模式选择，色度则进行基于亮度模式的选择，实现了预测流程的优化，降低了模式选择的复杂度，提高了模式选择的适用性。

作为一种可能的实现方式，确定待处理的图像帧PU单元的多种PU预测模式，包括：根据所述图像帧的帧类型确定所述图像帧PU单元的预测方式；所述预测方式包括帧内预测或帧间预测；根据所述图像帧的编码标准确定所述图像帧PU单元在所述预测方式下的所述多种PU预测模式。

在本申请实施例中，该预测模式的选择方法不仅可以应用于帧内预测，也可以应用于帧间预测。由于图像帧的帧类型有不同的情况，不同的帧类型都可以采用该选择方法，进一步提高了该选择方法的适用性。