chineseocr_lite
chineseocr_lite copied to clipboard
[W:onnxruntime:, graph.cc:2709 CleanUnusedInitializers] unused_initializer优化问题
运行时报onnxruntime warning
尝试使用 onnx.optimizer优化报错
有哪位大佬知道怎么搞吗,把warning关掉也行
需要修改模型数据。手动清理导出的模型,可以解决这个问题,参考下面的代码
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)