search

coolEphemeroptera / funasr_seaco_paraformer_onnx_with_timestamp

修复funasr中seaco-paraformer导出onnx后没有时间戳的bug
4 stars 2 forks source link

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

Open DrewdropLife opened 1 month ago

DrewdropLife commented 1 month ago

你好，想请教一下是否试过导出分角色识别的模型到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)

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

coolEphemeroptera commented 1 month ago

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

coolEphemeroptera commented 1 month ago

参考下这个https://github.com/coolEphemeroptera/3D-Speaker/blob/main/speakerlab/bin/export_speaker_embedding_onnx.py