昇腾社区首页
中文
注册

简化ONNX文件

ONNX Simplifier是一款开源工具,可以简化ONNX模型。通过推断整个计算图,用常量输出替换冗余运算符(也称为常量折叠)。

执行以下命令,使用ONNX Simplifier工具。

pip install onnx-simplifier
onnxsim  --overwrite-input-shape="1,3,224,24" efficient.onnx efficient_sim.onnx
onnxsim -h  #查看参数说明

图1所示,对导出的ONNX文件进行onnxsim后可以看到,减少了ONNX部分的操作。

图1 使用ONNX Simplifier工具