许可证：cc-by-4.0
语言：

• 英文
  基础模型：
• nvidia/parakeet-tdt-0.6b-v2
  任务标签：自动语音识别

NVIDIA Parakeet TDT 0.6B V2（英文版）模型已转换为ONNX格式，适用于onnx-asr。

安装onnx-asr

pip install onnx-asr[cpu,hub]  

加载Parakeet TDT模型并识别wav文件

import onnx_asr  
model = onnx_asr.load_model("nemo-parakeet-tdt-0.6b-v2")  
print(model.recognize("test.wav"))  

模型导出代码

import nemo.collections.asr as nemo_asr  
from pathlib import Path  

model = nemo_asr.models.ASRModel.from_pretrained("nvidia/parakeet-tdt-0.6b-v2")  

onnx_dir = Path("nemo-onnx")  
onnx_dir.mkdir(exist_ok=True)  
model.export(str(Path(onnx_dir, "model.onnx")))  

with Path(onnx_dir, "vocab.txt").open("wt") as f:  
    for i, token in enumerate([*model.tokenizer.vocab, "<blk>"]):  
        f.write(f"{token} {i}\n")