metadata
license: apache-2.0
FireRedASR-AED
小红书ASR AED-L版本在AX650N上的部署,原项目地址为:https://github.com/FireRedTeam/FireRedASR
转换后的模型放置在axmodel目录,目前支持中文、英文,最长输入10秒的音频,超过10秒的音频会用VAD切割后推理。
模型转换
支持平台
- AX650N
安装依赖
Audio backend
sudo apt install libsndfile1
Python
测试环境为Python 3.12,建议使用Miniconda,安装方法参考
conda create -n fireredasr python=3.12
conda activate fireredasr
pip install -r requirements.txt
requirements.txt 只包含 AX650N AED 推理路径需要的运行时依赖。其他脚本按需安装:
fireredasr_onnx.py:pip install onnxruntimefireredasr/speech2text.py的 LLM 路径:pip install transformers peftfireredasr/utils/wer.py --do_tn 1:pip install cn2an
手动安装pyaxengine
wget https://github.com/AXERA-TECH/pyaxengine/releases/download/0.1.3.rc2/axengine-0.1.3-py3-none-any.whl
pip install axengine-0.1.3-py3-none-any.whl
使用
conda activate fireredasr
python test_ax_model.py
hypo_axmodel.txt包含识别结果
性能表现
RTF ~= 0.3
CER(on custom dataset): 3.45%