未验证 提交 16d5a5f3 编写于 作者: D daquexian 提交者: GitHub

chore(tools): upgrade onnxsim to v0.4.1 (#1441)

chore(tools): upgrade onnxsim to v0.4.1
上级 33f48a92
......@@ -15,4 +15,4 @@ psutil
pycocotools>=2.0.2
onnx==1.8.1
onnxruntime==1.8.0
onnx-simplifier==0.3.5
onnx-simplifier==0.4.1
......@@ -102,16 +102,11 @@ def main():
if not args.no_onnxsim:
import onnx
from onnxsim import simplify
input_shapes = {args.input: list(dummy_input.shape)} if args.dynamic else None
# use onnxsimplify to reduce reduent model.
# use onnx-simplifier to reduce reduent model.
onnx_model = onnx.load(args.output_name)
model_simp, check = simplify(onnx_model,
dynamic_input_shape=args.dynamic,
input_shapes=input_shapes)
model_simp, check = simplify(onnx_model)
assert check, "Simplified ONNX model could not be validated"
onnx.save(model_simp, args.output_name)
logger.info("generated simplified onnx model named {}".format(args.output_name))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册