FunASR/funasr/export
2023-02-07 19:37:58 +08:00
..
models export model 2023-02-07 17:45:59 +08:00
utils export model 2023-02-07 17:45:59 +08:00
__init__.py export model 2023-02-07 15:19:18 +08:00
export_model.py export model 2023-02-07 19:37:58 +08:00
README.md export model 2023-02-07 19:32:24 +08:00
test_onnx.py export model 2023-02-07 17:52:29 +08:00

environment: ubuntu20.04-py37-torch1.11.0-tf1.15.5-1.2.0

Export onnx files from modelscope

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
export_model.export_from_modelscope('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export onnx files from local path

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
export_model.export_from_local('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')