chineseocr_lite icon indicating copy to clipboard operation
chineseocr_lite copied to clipboard

[W:onnxruntime:, graph.cc:2709 CleanUnusedInitializers] unused_initializer优化问题

Open terancejiang opened this issue 4 years ago • 1 comments

运行时报onnxruntime warning image 尝试使用 onnx.optimizer优化报错 image

有哪位大佬知道怎么搞吗,把warning关掉也行

terancejiang avatar Sep 27 '20 02:09 terancejiang

需要修改模型数据。手动清理导出的模型,可以解决这个问题,参考下面的代码

import onnx
import onnxoptimizer as optimizer
onnxfile = "chineseocr_lite-onnx\\models\\crnn_lite_lstm.onnx"
onnx_model = onnx.load(onnxfile)
passes = ["extract_constant_to_initializer", "eliminate_unused_initializer"]
optimized_model = optimizer.optimize(onnx_model, passes)
onnx.save(optimized_model, onnxfile)

lindatas avatar May 19 '21 10:05 lindatas