昇腾社区首页
中文
注册
开发者
下载
支持
积分兑换 NEW

venc_set_cu_pred

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

x

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

x

Atlas 训练系列产品

x

Atlas 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 200/300/500 推理产品

x

功能说明

本接口用于调节通道CU/MB的模式选择的倾向性，支持调节帧间预测（Inter）和帧内预测（Intra）的倾向性配置以及编码块大小的倾向性配置。

编码协议以像素块作为基本处理单元，H.265/H.264中分别称为CU(Coding Unit)/MB(MacroBlock)。本接口中的CU模式指Inter/Intra预测模式中对不同尺寸像素块的选择。

函数原型

  • C函数原型
    1
    hi_s32 hi_mpi_venc_set_cu_pred(hi_venc_chn chn, const hi_venc_cu_prediction *cu_pred)
  • python函数
    1
    ret = acl.himpi.venc_set_cu_pred(chn, cu_pred)

参数说明

参数名

说明

chn
int，编码通道号。
  • Atlas 推理系列产品，编码通道号的取值范围：[0, 256)，JPEGE功能和VENC功能共用通道，且通道总数最多256。
  • Atlas 200I/500 A2 推理产品，编码通道号的取值范围：[0, 128)，JPEGE功能和VENC功能共用通道，且通道总数最多128。

cu_pred

dict，CU模式选择的倾向性参数字典。具体参见hi_venc_cu_prediction。

该字典内需包含以下参数：
  • pred_mode：倾向性选择模式，支持auto和manual两种模式。auto模式下由驱动内部完成倾向性配置，即用户的配置参数不生效；manual模式允许用户根据实际场景完成自定义的倾向性配置。
  • intra32_cost/intra16_cost/intra8_cost/intra4_cost/inter64_cost/inter32_cost/inter16_cost/inter8_cost分别设置各自模式的倾向性大小。默认每个cost值为8，即不做倾向性配置。每个模式的cost设置越大意味着选择该模式的倾向性越小。

返回值说明

返回值

说明

ret

int，错误码。

约束说明

  • 仅H.265/H.264编码通道支持。
  • 建议用户在创建通道之后，启动编码之前调用本接口，减少编码过程中调用本接口的次数。若在编码过程中调用本接口，则下一帧编码时配置生效。
  • 建议用户在调用此接口之前先调用acl.himpi.venc_get_cu_pred接口，获取CU模式选择的倾向性配置，然后再进行设置。
  • 注意某些场景下特定模式不使能（例如，H.264中最大MB尺寸为16*16，且baseline和main profile下inter16_cost不使能，high profile下inter8_cost不使能），即使用户设置了这些特定模式的倾向性参数也不生效。