Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

分角色识别的能否支持onnx导出? #1

Open
DrewdropLife opened this issue May 14, 2024 · 2 comments
Open

分角色识别的能否支持onnx导出? #1

DrewdropLife opened this issue May 14, 2024 · 2 comments

Comments

@DrewdropLife
Copy link

DrewdropLife commented May 14, 2024

你好,想请教一下是否试过导出分角色识别的模型到onnx?

from funasr.auto.auto_model import AutoModel

model = AutoModel(model="paraformer-zh", 
                  vad_model="fsmn-vad", 
                  punc_model="ct-punc-c", 
                  spk_model="cam++", 
                  )
res = model.export(quantize=False)

不知道应该修改什么地方呢

@pika-online
Copy link
Owner

funasr目前没有制作cam++的onnx-graph,需要自己手动导出torch模型到onnx, https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/models/campplus

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants