File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change 1515pycocotools >= 2.0.2
1616onnx == 1.8.1
1717onnxruntime == 1.8.0
18- onnx-simplifier == 0.3.5
18+ onnx-simplifier == 0.4.1
Original file line number Diff line number Diff line change @@ -102,16 +102,11 @@ def main():
102102
103103 if not args .no_onnxsim :
104104 import onnx
105-
106105 from onnxsim import simplify
107106
108- input_shapes = {args .input : list (dummy_input .shape )} if args .dynamic else None
109-
110- # use onnxsimplify to reduce reduent model.
107+ # use onnx-simplifier to reduce reduent model.
111108 onnx_model = onnx .load (args .output_name )
112- model_simp , check = simplify (onnx_model ,
113- dynamic_input_shape = args .dynamic ,
114- input_shapes = input_shapes )
109+ model_simp , check = simplify (onnx_model )
115110 assert check , "Simplified ONNX model could not be validated"
116111 onnx .save (model_simp , args .output_name )
117112 logger .info ("generated simplified onnx model named {}" .format (args .output_name ))
You can’t perform that action at this time.
0 commit comments