From 5016f29452ce6705e038d1484044404a45b7b54c Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期三, 29 三月 2023 15:54:04 +0800 Subject: [PATCH] export --- funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/top_level.txt | 1 funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/SOURCES.txt | 17 ++++++++ funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/dependency_links.txt | 1 funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/requires.txt | 7 +++ funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/PKG-INFO | 80 ++++++++++++++++++++++++++++++++++++++++ 5 files changed, 106 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/PKG-INFO b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/PKG-INFO new file mode 100644 index 0000000..94d2cb8 --- /dev/null +++ b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/PKG-INFO @@ -0,0 +1,80 @@ +Metadata-Version: 2.1 +Name: funasr-onnx +Version: 0.0.3 +Summary: FunASR: A Fundamental End-to-End Speech Recognition Toolkit +Home-page: https://github.com/alibaba-damo-academy/FunASR.git +Author: Speech Lab, Alibaba Group, China +Author-email: funasr@list.alibaba-inc.com +License: MIT +Keywords: funasr,asr +Platform: Any +Classifier: Programming Language :: Python :: 3.6 +Classifier: Programming Language :: Python :: 3.7 +Classifier: Programming Language :: Python :: 3.8 +Classifier: Programming Language :: Python :: 3.9 +Classifier: Programming Language :: Python :: 3.10 +Description-Content-Type: text/markdown + +## Using funasr with ONNXRuntime + + +### Introduction +- Model comes from [speech_paraformer](https://www.modelscope.cn/models/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/summary). + + +### Steps: +1. Export the model. + - Command: (`Tips`: torch >= 1.11.0 is required.) + + More details ref to ([export docs](https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/export)) + + - `e.g.`, Export model from modelscope + ```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 False + ``` + - `e.g.`, Export model from local path, the model'name must be `model.pb`. + ```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 False + ``` + + +2. Install the `funasr_onnx` + +install from pip +```shell +pip install --upgrade funasr_onnx -i https://pypi.Python.org/simple +``` + +or install from source code + +```shell +git clone https://github.com/alibaba/FunASR.git && cd FunASR +cd funasr/runtime/python/funasr_onnx +python setup.py build +python setup.py install +``` + +3. Run the demo. + - Model_dir: the model path, which contains `model.onnx`, `config.yaml`, `am.mvn`. + - Input: wav formt file, support formats: `str, np.ndarray, List[str]` + - Output: `List[str]`: recognition result. + - Example: + ```python + from funasr_onnx import Paraformer + + model_dir = "/nfs/zhifu.gzf/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" + model = Paraformer(model_dir, batch_size=1) + + wav_path = ['/nfs/zhifu.gzf/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/example/asr_example.wav'] + + result = model(wav_path) + print(result) + ``` + +## Performance benchmark + +Please ref to [benchmark](https://github.com/alibaba-damo-academy/FunASR/blob/main/funasr/runtime/python/benchmark_onnx.md) + +## Acknowledge +1. This project is maintained by [FunASR community](https://github.com/alibaba-damo-academy/FunASR). +2. We acknowledge [SWHL](https://github.com/RapidAI/RapidASR) for contributing the onnxruntime (for paraformer model). diff --git a/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/SOURCES.txt b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/SOURCES.txt new file mode 100644 index 0000000..e759e27 --- /dev/null +++ b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/SOURCES.txt @@ -0,0 +1,17 @@ +README.md +setup.py +funasr_onnx/__init__.py +funasr_onnx/paraformer_bin.py +funasr_onnx/punc_bin.py +funasr_onnx/vad_bin.py +funasr_onnx.egg-info/PKG-INFO +funasr_onnx.egg-info/SOURCES.txt +funasr_onnx.egg-info/dependency_links.txt +funasr_onnx.egg-info/requires.txt +funasr_onnx.egg-info/top_level.txt +funasr_onnx/utils/__init__.py +funasr_onnx/utils/e2e_vad.py +funasr_onnx/utils/frontend.py +funasr_onnx/utils/postprocess_utils.py +funasr_onnx/utils/timestamp_utils.py +funasr_onnx/utils/utils.py \ No newline at end of file diff --git a/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/dependency_links.txt b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/dependency_links.txt new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/dependency_links.txt @@ -0,0 +1 @@ + diff --git a/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/requires.txt b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/requires.txt new file mode 100644 index 0000000..6fcb632 --- /dev/null +++ b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/requires.txt @@ -0,0 +1,7 @@ +librosa +onnxruntime>=1.7.0 +scipy +numpy>=1.19.3 +typeguard +kaldi-native-fbank +PyYAML>=5.1.2 diff --git a/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/top_level.txt b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/top_level.txt new file mode 100644 index 0000000..de41eb9 --- /dev/null +++ b/funasr/runtime/python/onnxruntime/funasr_onnx.egg-info/top_level.txt @@ -0,0 +1 @@ +funasr_onnx -- Gitblit v1.9.1