函数:get_input_dims_v2
C函数原型 |
aclError aclmdlGetInputDimsV2(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims) |
---|---|
Python函数 |
dims, ret = acl.mdl.get_input_dims_v2(model_desc, index) |
函数功能 |
根据模型描述获取模型的输入Tensor的维度信息。 如果模型中含有静态AIPP配置信息,您可以根据实际需要选择acl.mdl.get_input_dims接口或acl.mdl.get.input_dims_v2接口查询维度信息,两者的区别在于: |
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输入的大小,index值从0开始。 |
返回值说明 |
dims:dict,针对多batch或多种宽高的场景,按照该字典的数组返回其支持的最大档的组合,具体请参见aclmdlIODims。 dims = { "name": xxx, #tensor name "dimCount":xxx,#shape中的维度个数 "dims": [xx, xx, xx] # 维度信息 } 针对动态Batch、动态分辨率(宽高)的场景,输入Tensor的“dims”中Batch数或宽高为-1,表示其动态可变。 例如,输入Tensor的“format”为NCHW,在动态Batch场景下,动态可变的输入Tensor的“dims”为[-1, 3, 224, 224],在动态分辨率场景下,动态可变的输入Tensor的“dims”为[1, 3, -1, -1]。举例中的斜体部分以实际情况为准。
针对静态AIPP场景,本接口针对不同格式的图像,对应NHWC的Format格式,当前接口中明确各个维度的定义规则,如表1所示。 ret:int,错误码。
|
约束说明 |
无 |