From 7584bbd6f3e321cc8bc970739a7cfce29ffcc18b Mon Sep 17 00:00:00 2001 From: haoneng.lhn <haoneng.lhn@alibaba-inc.com> Date: 星期四, 27 四月 2023 00:21:20 +0800 Subject: [PATCH] update paraformer streaming code --- funasr/runtime/onnxruntime/readme.md | 79 ++++++++++++++++++++++----------------- 1 files changed, 45 insertions(+), 34 deletions(-) diff --git a/funasr/runtime/onnxruntime/readme.md b/funasr/runtime/onnxruntime/readme.md index dddb46a..95840e5 100644 --- a/funasr/runtime/onnxruntime/readme.md +++ b/funasr/runtime/onnxruntime/readme.md @@ -1,33 +1,23 @@ +# ONNXRuntime-cpp -## Demo -```shell -tester /path/models_dir /path/wave_file quantize(true or false) -``` - -The structure of /path/models_dir -``` -config.yaml, am.mvn, model.onnx(or model_quant.onnx) -``` - -## Steps - -### Export onnx -#### Install [modelscope and funasr](https://github.com/alibaba-damo-academy/FunASR#installation) +## Export the model +### Install [modelscope and funasr](https://github.com/alibaba-damo-academy/FunASR#installation) ```shell pip3 install torch torchaudio pip install -U modelscope pip install -U funasr ``` -#### Export [onnx model](https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/export) + +### Export [onnx model](https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/export) ```shell python -m funasr.export.export_model --model-name damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch --export-dir ./export --type onnx --quantize True ``` -### Building for Linux/Unix +## Building for Linux/Unix -#### Download onnxruntime +### Download onnxruntime ```shell # download an appropriate onnxruntime from https://github.com/microsoft/onnxruntime/releases/tag/v1.14.0 # here we get a copy of onnxruntime for linux 64 @@ -35,36 +25,57 @@ tar -zxvf onnxruntime-linux-x64-1.14.0.tgz ``` -#### Install fftw3 -```shell -sudo apt install libfftw3-dev #ubuntu -# sudo yum install fftw fftw-devel #centos -``` - -#### Install openblas +### Install openblas ```shell sudo apt-get install libopenblas-dev #ubuntu # sudo yum -y install openblas-devel #centos ``` -#### Build runtime +### Build runtime ```shell git clone https://github.com/alibaba-damo-academy/FunASR.git && cd funasr/runtime/onnxruntime mkdir build && cd build cmake -DCMAKE_BUILD_TYPE=release .. -DONNXRUNTIME_DIR=/path/to/onnxruntime-linux-x64-1.14.0 make ``` +## Run the demo -#### The structure of a qualified onnxruntime package. +```shell +./funasr-onnx-offline [--wav-scp <string>] [--wav-path <string>] + [--punc-config <string>] [--punc-model <string>] + --am-config <string> --am-cmvn <string> + --am-model <string> [--vad-config <string>] + [--vad-cmvn <string>] [--vad-model <string>] [--] + [--version] [-h] +Where: + --wav-scp <string> + wave scp path + --wav-path <string> + wave file path + + --punc-config <string> + punc config path + --punc-model <string> + punc model path + + --am-config <string> + (required) am config path + --am-cmvn <string> + (required) am cmvn path + --am-model <string> + (required) am model path + + --vad-config <string> + vad config path + --vad-cmvn <string> + vad cmvn path + --vad-model <string> + vad model path + + Required: --am-config <string> --am-cmvn <string> --am-model <string> + If use vad, please add: [--vad-config <string>] [--vad-cmvn <string>] [--vad-model <string>] + If use punc, please add: [--punc-config <string>] [--punc-model <string>] ``` -onnxruntime_xxx -鈹溾攢鈹�鈹�include -鈹斺攢鈹�鈹�lib -``` - -### Building for Windows - -Ref to win/ ## Acknowledge 1. This project is maintained by [FunASR community](https://github.com/alibaba-damo-academy/FunASR). -- Gitblit v1.9.1