--input_format
功能说明
指定模型输入数据格式。
关联参数
无。
参数取值
参数值
:
当原始框架为Caffe时,支持NCHW、ND(表示支持任意格式,N<=4)两种格式,默认为NCHW。
当原始框架是TensorFlow时,支持NCHW、NHWC、ND、NCDHW、NDHWC五种输入格式,默认为NHWC。
当原始框架为ONNX时,支持NCHW、NCDHW、ND(表示支持任意格式,N<=4)三种格式,默认为NCHW。
参数默认值
:Caffe、ONNX默认为NCHW;TensorFlow默认为NHWC。
推荐配置及收益
无。
示例
--input_format=NCHW
依赖约束
无。
父主题:
输入选项