--input_shape_range
功能说明
指定模型输入数据的shape范围。该参数预留,当前版本暂不支持。
关联参数
该参数不能与--dynamic_batch_size、--dynamic_image_size、--dynamic_dims同时使用。
参数取值
参数值:模型输入数据的shape范围信息,例如:"input_name1:[n1~n2,c1,h1,w1];input_name2:[n2,c2,h2,w2]"。指定的节点必须放在双引号中,节点中间使用英文分号分隔。input_name必须是转换前的网络模型中的节点名称。
参数值约束:
- shape范围信息必须放在英文[]中。
- 该参数不限定维度,维度中的任一值都可以由用户指定取值范围。
- 如果用户不想指定维度的取值,则可以将其设置为-1,表示此维度可以使用>=1的任意取值。
推荐配置及收益
无。
示例
--input_shape_range="input1:[8~20,3,5,-1];input2:[5,3~9,10,-1]"
支持的芯片型号
昇腾310P AI处理器
昇腾910 AI处理器
依赖约束
- 使用约束:
- 该参数只适用于TensorFlow和ONNX网络模型,不适用于Caffe和MindSpore网络模型。
- 若使用该参数时,同时通过--insert_op_conf设置了AIPP功能,则AIPP输出图片的宽和高要在--input_shape_range所设置的范围内。
- 接口约束:
如果模型转换时通过该参数设置了shape的范围,则使用应用工程进行模型推理时,需要在aclmdlExecute接口之前,调用aclmdlSetDatasetTensorDesc接口,用于设置真实的输入Tensor描述信息(输入shape范围);模型执行之后,调用aclmdlGetDatasetTensorDesc接口获取模型动态输出的Tensor描述信息;再进一步调用aclTensorDesc下的操作接口获取输出Tensor数据占用的内存大小、Tensor的Format信息、Tensor的维度信息等)。
关于aclmdlSetDatasetTensorDesc、aclmdlGetDatasetTensorDesc等接口的具体使用方法,请参见《应用软件开发指南 (C&C++)》手册“AscendCL API参考”。
父主题: 输入选项