--input_format

功能说明

指定模型输入数据的格式。

关联参数

无。

参数取值

参数值：

• 当原始框架为Caffe时，支持NCHW、ND（表示支持任意维度格式，N<=4）两种格式，默认为NCHW。
• 当原始框架为ONNX时，支持NCHW、NCDHW、ND（表示支持任意维度格式，N<=4）三种格式，默认为NCHW。
• 当原始框架是TensorFlow时，支持NCHW、NHWC、ND、NCDHW、NDHWC五种输入格式，默认为NHWC。
  • 如果TensorFlow模型是通过ONNX模型转换工具输出的，则该参数必填，且值为NCHW。
  • 如果原始模型中含有带data_format入参的算子，则该参数必填，推荐取值为ND，模型转换过程中会根据data_format属性的算子，推导出具体的format。若用户无法确定输入数据格式，则推荐指定为ND。
• 当原始框架为MindSpore时，只支持配置为NCHW，设置为其它值无效，但模型转换成功。

参数默认值：Caffe、MindSpore、ONNX默认为NCHW；TensorFlow默认为NHWC。

参数值约束：

• 如果模型转换时开启AIPP，在进行推理业务时，输入图片数据要求为NHWC排布，该场景下最终与AIPP连接的输入节点的格式被强制改成NHWC，可能与atc模型转换命令中--input_format参数指定的格式不一致。
• 如果同时配置了--insert_op_conf参数，则--input_format参数只能配置为NCHW、NHWC。
• 该参数不支持模型多输入且不同输入存在不同数据格式的设置场景，即该参数只支持设置为单个取值，例如TensorFlow框架的模型，只支持设置为NCHW、NHWC、ND、NCDHW、NDHWC五种输入格式中的一种。

推荐配置及收益

无。

示例

--input_format=NCHW

支持的芯片型号

昇腾310B AI处理器

依赖约束

无。