函数:set_dynamic_hw_size
产品支持情况
产品 |
是否支持 |
---|---|
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
动态分辨率场景下,在模型执行前调用本接口设置模型推理时输入图片的高和宽。
函数原型
- C函数原型
1
aclError aclmdlSetDynamicHWSize(uint32_t modelId, aclmdlDataset *dataset, size_t index, uint64_t height, uint64_t width)
- python函数
1
ret = acl.mdl.set_dynamic_hw_size(model_id, dataset, index, height, width)
参数说明
参数名 |
说明 |
---|---|
model_id |
int,模型ID。 可在以下接口成功加载模型后获取到模型ID。 |
dataset |
int,表示模型的输入数据。 使用aclmdlDataset类型的数据描述模型推理时的输入数据,输入的内存地址、内存大小用aclDataBuffer类型的数据来描述,具体请参见aclmdlDataset。 |
index |
int,标识动态Batch输入的输入index,需调用acl.mdl.get_input_index_by_name接口获取,动态Batch和动态分辨率输入的名称固定为“ascend_mbatch_shape_data”,动态AIPP输入的名称固定为“ascend_dynamic_aipp_data”。 |
height |
int,需设置的H值。 |
width |
int,需设置的W值。 此处设置的分辨率(“height”与“width”)只能为模型构建时设置的分辨率档位中的其中一档,也可以调用acl.mdl.get_dynamic_hw接口获取指定模型支持的分辨率档位数以及每一档中的宽、高。 |
返回值说明
返回值 |
说明 |
---|---|
ret |
int,错误码,返回0表示成功,返回其它值表示失败。 |