You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
PaddleSpeech/speechx/examples/ds2_ol/onnx
Hui Zhang 59a78f2a46
ds2 wenetspeech to onnx and support streaming asr server
2 years ago
..
local ds2 wenetspeech to onnx and support streaming asr server 2 years ago
.gitignore test with run.sh 2 years ago
README.md ds2 wenetspeech to onnx and support streaming asr server 2 years ago
path.sh pdmodel prune and infershape 2 years ago
run.sh ds2 wenetspeech to onnx and support streaming asr server 2 years ago
utils pdmodel prune and infershape 2 years ago

README.md

DeepSpeech2 ONNX model

  1. convert deepspeech2 model to ONNX, using Paddle2ONNX.
  2. check paddleinference and onnxruntime output equal.
  3. optimize onnx model
  4. check paddleinference and optimized onnxruntime output equal.

Please make sure Paddle2ONNX and onnx-simplifier version is correct.

The example test with these packages installed:

paddle2onnx              0.9.8    # develop 62c5424e22cd93968dc831216fc9e0f0fce3d819
paddleaudio              0.2.1
paddlefsl                1.1.0
paddlenlp                2.2.6
paddlepaddle-gpu         2.2.2
paddlespeech             0.0.0       # develop
paddlespeech-ctcdecoders 0.2.0
paddlespeech-feat        0.1.0
onnx                     1.11.0
onnx-simplifier          0.0.0       # https://github.com/zh794390558/onnx-simplifier/tree/dyn_time_shape
onnxoptimizer            0.2.7
onnxruntime              1.11.0

Using

bash run.sh

For more details please see run.sh.

Outputs

The optimized onnx model is exp/model.opt.onnx.

To show the graph, please using local/netron.sh.