aclmdlGetInputDimsV2

函数功能

根据模型描述信息获取模型的输入tensor的维度信息。同步接口。

如果模型中含有静态AIPP配置信息,您可以根据实际需要选择aclmdlGetInputDims接口或aclmdlGetInputDimsV2接口查询维度信息,两者的区别在于:

函数原型

aclError aclmdlGetInputDimsV2(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims)

参数说明

参数名

输入/输出

说明

modelDesc

输入

aclmdlDesc类型的指针。

需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。

index

输入

指定获取第几个输入的Dims,index值从0开始。

dims

输出

输入维度信息的指针。

针对动态Batch、动态分辨率(宽高)的场景,输入tensor的dims中batch size或宽高为-1,表示其动态可变。例如,输入tensor的format为NCHW,在动态Batch场景下,动态可变的输入tensor的dims为[-1,3,224,224];在动态分辨率场景下,动态可变的输入tensor的dims为[1,3,-1,-1]。举例中的斜体部分以实际情况为准。

若tensor的name长度大于127,则在输出的dims.name时,AscendCL会将tensor的name转换为“acl_modelId_${id}_input_${index}_${随机字符串} ”格式(如果转换后的tensor的name与模型中已有的tensor的name冲突,则会在转换后的name尾部增加“_${随机字符串} ”,否则不会增加随机字符串),并在转换后的name与原name之间建立映射关系,用户可调用aclmdlGetTensorRealName接口,传入转换后的name,获取原name(若向接口传入原name,则获取的还是原name);若tensor的name长度小于或等于127,则在输出的dims.name时,按tensor的name输出。

针对静态AIPP场景,本接口针对不同格式的图像,对应NHWC的Format格式,当前接口中明确各个维度的定义规则,如表1所示。

表1 静态AIPP场景下的维度定义规则

图像格式

Format参考格式

维度定义规则

YUV420SP_U8

NHWC

n,h*1.5,w,1

XRGB8888_U8

NHWC

n,h,w,4

RGB888_U8

NHWC

n,h,w,3

YUV400_U8

NHWC

n,h,w,1

ARGB8888_U8

NHWC

n,h,w,4

YUYV_U8

NHWC

n,h,w,2

YUV422SP_U8

NHWC

n,h*2,w,1

AYUV444_U8

NHWC

n,h,w,4

返回值说明

返回0表示成功,返回非0表示失败。