From 837dc3758a5364fd720bb44f497c82aebe4f7dab Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 21 二月 2023 16:36:15 +0800
Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add
---
funasr/runtime/python/onnxruntime/README.md | 28 ++++++++++++++++------------
1 files changed, 16 insertions(+), 12 deletions(-)
diff --git a/funasr/runtime/python/onnxruntime/README.md b/funasr/runtime/python/onnxruntime/README.md
index 38414d5..cfa9f36 100644
--- a/funasr/runtime/python/onnxruntime/README.md
+++ b/funasr/runtime/python/onnxruntime/README.md
@@ -10,16 +10,7 @@
### Steps:
-1. Download the whole directory
-```shell
-git clone https://github.com/alibaba/FunASR.git && cd FunASR
-cd funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer
-```
-2. Install the related packages.
- ```bash
- pip install -r requirements.txt
- ```
-3. Export the model.
+1. Export the model.
`Tips`: torch 1.11.0 is required.
@@ -42,13 +33,26 @@
python -m funasr.export.export_model '/mnt/workspace/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" true
```
-5. Run the demo.
+
+2. Install the `rapid_paraformer`
+ - Build the rapid_paraformer `whl`
+ ```shell
+ git clone https://github.com/alibaba/FunASR.git && cd FunASR
+ cd funasr/runtime/python/onnxruntime/rapid_paraformer
+ python setup.py bdist_wheel
+ ```
+ - Install the build `whl`
+ ```bash
+ pip install dist/rapid_paraformer-0.0.1-py3-none-any.whl
+ ```
+
+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 paraformer_onnx import Paraformer
+ from rapid_paraformer 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)
--
Gitblit v1.9.1