概览

总体选项

--help或--h

可选

打印帮助信息。

输入选项

--model

必选

原始模型文件路径与文件名。

--weights

--framework为Caffe时，必选。

权重文件路径与文件名。

当原始模型是Caffe时需要指定。

--framework

必选

原始框架类型。

当前只支持配置为0（Caffe）、3（TensorFlow）或者5（ONNX）。

--input_format

可选

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

TensorFlow默认为NHWC，Caffe默认为NCHW，ONNX默认为NCHW。

--calibration_data

必选

模型的校准数据文件。

--calibration_shape

必选

模型的校准模型输入节点名称和shape信息。

--calibration_config

可选

简易配置文件，参数为空，则使用默认配置。

输出选项

--output

必选

输出的air文件名，会自动添加".air"后缀

目标芯片类型

--soc_version

必选

当前只支持配置为AscendxxxA和AscendxxxB。

功能配置选项

--out_nodes

可选

指定输出节点。

--input_fp16_nodes

可选

设置某个输入是否为fp16类型。

调试选项

--log

可选

设置日志级别。