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

权利要求书

1.一种视频码率控制方法，其特征在于，包括：

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

B、利用预设滤波核对所述视频I帧中每个编码树单元的内容特征进行提取，所述内容特征包括能量特征、强度特征、形状特征以及平滑性特征；所述预设滤波核包括方向不同的四个滤波核，方向分别为0度、45度、90度和135度；

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

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

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

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

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

将所述四个滤波核分别与所述视频I帧中的每个编码树单元进行卷积操作，得到不同方向的四个特征矩阵；

根据所述四个特征矩阵以及第一公式确定所述能量特征，所述第一公式为：

其中，G_f1表示所述能量特征，G_CTU(θ,i,j)表示在θ方向上的特征矩阵中位置为(i，j)的灰度值，X＝[0°，45°，90°，135°]，H表示编码树单元的高度，W表示编码树单元的宽度，N表示编码树单元的像素个数，ω₁＝0.24，ω₂＝0.0059，C₁＝-2.4；

根据所述四个特征矩阵以及第二公式确定所述强度特征，所述第二公式为：

其中，G_f2表示所述强度特征，ω₃＝1.3；

根据所述四个特征矩阵以及第三公式确定所述形状特征，所述第三公式为：

其中，G_f3表示所述形状特征，ω₄＝0.85；

根据所述四个特征矩阵、第四公式以及第五公式确定所述平滑性特征，所述第四公式和所述第五公式分别为：

其中，G_v(θ)表示编码树单元在θ方向上的平滑性特征，G_f4表示编码树单元在四个方向上的平滑性特征，表示在θ方向上的特征矩阵的平均值，α₁＝2.0，α₂＝1.9，ω₅＝0.97，ω₆＝4.2，ω₇＝2.7，C₂＝998；

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

其中，表示所述当前编码树单元的实际码率，k表示所述视频I帧中的第k个编码树单元，GC(k)表示所述当前编码树单元的复杂度，表示从所述当前编码树单元到所述视频I帧的最后一个编码树单元的复杂度的总和，l表示所述视频I帧中的第l个编码树单元，M表示所述视频I帧中编码树单元的总数，表示所述当前剩余码率，C₃＝0.13，C₄＝0.24，ω₈＝4.3，α₃＝1.5，α₄＝2，α₅＝0.5，α₆＝2.02。