[发明专利]编码单元深度确定方法及装置

权利要求书

1.一种编码单元深度确定方法，其特征在于，包括：

确定待处理编码单元的当前最优模式的残差系数；

在所述残差系数不为零时，若所述待处理编码单元的编码深度为零，则从所述待处理编码单元及所述待处理编码单元所在编码树单元的近邻编码树单元中，分别获取设定类型的编码信息特征，组成预测特征向量样本；

将所述预测特征向量样本输入预训练的预测模型中，得到所述预测模型输出的预测结果，所述预测结果表明所述待处理编码单元是否需要进一步进行深度划分；

若所述待处理编码单元的编码深度不为零，则确定所述待处理编码单元所在编码树单元的近邻编码树单元中，与所述待处理编码单元相同编码深度的编码单元的平均代价，作为第一平均代价；确定所述待处理编码单元所在编码树单元中相同编码深度的已编码的编码单元的平均代价，作为第二平均代价；根据所述第一平均代价及所述第二平均代价，确定代价阈值；若所述待处理编码单元的当前最优模式的代价小于所述代价阈值，则确定所述待处理编码单元不需要进一步进行深度划分；若所述待处理编码单元的当前最优模式的代价不小于所述代价阈值，则确定所述待处理编码单元需要进一步进行深度划分；

其中，所述预测模型为利用标记有分类结果的训练样本预训练得到，所述训练样本包括所述设定类型的编码信息特征。

2.根据权利要求1所述的方法，其特征在于，所述待处理编码单元属于非I帧视频图像。

3.根据权利要求1-2任一项所述的方法，其特征在于，所述预测模型包括P帧预测模型和B帧预测模型，所述P帧预测模型预训练时使用的训练样本为，属于P帧视频图像的编码单元中提取的所述设定类型的编码信息特征，所述B帧预测模型预训练时使用的训练样本为，属于B帧视频图像的编码单元中提取的所述设定类型的编码信息特征；

所述将所述预测特征向量样本输入预训练的预测模型中，得到所述预测模型输出的预测结果，包括：

确定所述待处理编码单元所属视频帧图像的类型为P帧还是B帧；

若为P帧，则将所述预测特征向量样本输入所述P帧预测模型，得到所述P帧预测模型输出的预测结果；

若为B帧，则将所述预测特征向量样本输入所述B帧预测模型，得到所述B帧预测模型输出的预测结果。

4.根据权利要求1-2任一项所述的方法，其特征在于，所述从所述待处理编码单元及所述待处理编码单元所在编码树单元的近邻编码树单元中，分别获取设定类型的编码信息特征，包括：

获取所述待处理编码单元的代价、量化系数、失真及方差；

获取所述待处理编码单元所在编码树单元的近邻编码树单元的代价和深度信息。

5.根据权利要求1所述的方法，其特征在于，所述确定所述待处理编码单元所在编码树单元的近邻编码树单元中，与所述待处理编码单元相同编码深度的编码单元的平均代价，作为第一平均代价，包括：

针对所述待处理编码单元所在编码树单元的每一近邻编码树单元，确定其中与所述待处理编码单元相同编码深度的编码单元的平均代价；

按照每一所述近邻编码树单元与所述待处理编码单元所在编码树单元的方位关系，确定每一所述近邻编码树单元的权重值；

根据每一所述近邻编码树单元的权重值及其平均代价，确定各所述近邻编码树单元的加权平均代价，作为第一平均代价。