[发明专利]一种视频码率控制方法、装置、计算机设备及存储介质

说明书

本发明实施例公开了一种视频码率控制方法、装置、计算机设备及存储介质。该方法包括：获取视频I帧及其信息特征；利用预设滤波核对视频I帧中每个编码树单元的内容特征进行提取；根据当前编码树单元的内容特征以及视频I帧的当前剩余码率确定为其分配的实际码率；逐个将当前编码树的内容特征、实际码率、当前剩余码率以及视频I帧的信息特征汇中的量化参数输入训练好的学习模型中，以预测当前编码树单元的量化参数；根据该量化参数对当前编码树单元进行编码。本发明实施例所提供的技术方案，通过分析视频的内容特征，并以此来完成码率的分配，再利用学习模型对量化参数进行预测，有效的提升了码率控制的准确性，从而提升了视频的视觉质量。

技术领域

本发明实施例涉及视频编码技术领域，尤其涉及一种视频码率控制方法、装置、计算机设备及存储介质。

背景技术

在视频传输的过程中信道的带宽往往是受限的，如何能充分利用有限的带宽资源编码视频并且保证比较良好的视频质量是非常重要的问题。因此，码率控制在这个场景下起着十分重要的作用。码率控制的目的就是在码率有限的情况下，合理使用比特进行编码，使视频压缩后的失真最小。在视频编码中，视频帧的类型主要分为I帧、P帧和B帧三类，而其中I帧的码率控制通常是最重要的。由于I帧需要作为参考帧给后续编码的视频帧作为参考图像，因此I帧需要消耗更多的码率以提升图像质量，如果I帧使用了过多的比特，会造成后续视频帧的码率缺失，另一方面，如果使用低质量的I帧作为参考帧，则会降低视频的质量。因此，平衡I帧的码率与失真是一个关键的问题。

现有技术中提出了采用R-λ模型建立的码率控制算法模型，并突出采用额SATD作为评估一个块的复杂程度，由于该码率控制算法模型是通过统计特性而凭经验所建立的，因此准确性较低，而且，现有的复杂度评估方法SATD也并不能有效的反映图像内容的特征。

发明内容

本发明实施例提供一种视频码率控制方法、装置、计算机设备及存储介质，以实现提升码率控制的准确性，从而提升视频的视觉质量。

第一方面，本发明实施例提供了一种视频码率控制方法，该方法包括：

A、获取视频I帧以及所述视频I帧的信息特征，所述信息特征包括所述视频I帧的量化参数；

B、利用预设滤波核对所述视频I帧中每个编码树单元的内容特征进行提取，所述内容特征包括能量特征、强度特征、形状特征以及平滑性特征；

C、根据当前编码树单元的所述内容特征以及所述视频I帧的当前剩余码率确定为所述当前编码树单元分配的实际码率；

D、将所述当前编码树单元的所述内容特征、所述实际码率、所述当前剩余码率以及所述视频I帧的量化参数输入训练好的学习模型中，以对所述当前编码树单元的量化参数进行预测；

E、根据所述当前编码树单元的量化参数对所述当前编码树单元进行编码；

F、判断所述当前编码树单元是否为所述视频I帧中的最后一个编码树单元；若是，则结束所述视频I帧的编码过程；若否，则将所述当前编码树单元更新为后一编码树单元并返回执行步骤C。

第二方面，本发明实施例还提供了一种视频码率控制装置，该装置包括：

信息特征获取模块，用于获取视频I帧以及所述视频I帧的信息特征，所述信息特征包括所述视频I帧的量化参数；

内容特征提取模块，用于利用预设滤波核对所述视频I帧中每个编码树单元的内容特征进行提取，所述内容特征包括能量特征、强度特征、形状特征以及平滑性特征；

实际码率分配模块，用于根据当前编码树单元的所述内容特征以及所述视频I帧的当前剩余码率确定为所述当前编码树单元分配的实际码率；