Automatic speech recognition toolkit on Pytorch
Project description
MASR流式与非流式语音识别
MASR是一款基于Pytorch实现的自动语音识别框架,MASR全称是神奇的自动语音识别框架(Magical Automatic Speech Recognition),MASR致力于简单,实用的语音识别项目。可部署在服务器,Nvidia Jetson设备,未来还计划支持Android等移动设备。
欢迎大家扫码入QQ群讨论,或者直接搜索QQ群号1169600237
,问题答案为博主Github的IDyeyupiaoling
。
本项目使用的环境:
- Anaconda 3
- Python 3.7
- Pytorch 1.10.0
- Windows 10 or Ubuntu 18.04
更新记录
- 2022.08.22: 增加非流式模型
deepspeech2_no_stream
和deepspeech2_big_no_stream
。 - 2022.08.04: 发布1.0版本,优化实时识别流程。
- 2022.07.12: 完成GUI界面的录音实时识别。
- 2022.06.14: 支持
deepspeech2_big
模型,适合WenetSpeech大数据集训练模型。 - 2022.01.16: 支持多种预处理方法。
- 2022.01.15: 支持英文语音识别。
- 2022.01.13: 支持给识别结果加标点符号
- 2021.12.26: 支持pip方式安装。
- 2021.12.25: 初步完成基本程序。
模型下载
本项目支持流式识别模型deepspeech2
、deepspeech2_big
,非流式模型deepspeech2_no_stream
、deepspeech2_big_no_stream
。
使用模型 | 数据集 | 预处理方式 | 语言 | 测试集字错率(词错率) | 下载地址 |
---|---|---|---|---|---|
deepspeech2 | aishell (179小时) | linear | 中文 | 0.06346 | 点击下载 |
deepspeech2_big | aishell (179小时) | linear | 中文 | ||
deepspeech2_no_stream | aishell (179小时) | linear | 中文 | ||
deepspeech2_big_no_stream | aishell (179小时) | linear | 中文 | ||
deepspeech2 | Librispeech (960小时) | linear | 英文 | 0.12842 | 点击下载 |
deepspeech2_big | Librispeech (960小时) | linear | 英文 | ||
deepspeech2_no_stream | Librispeech (960小时) | linear | 英文 | ||
deepspeech2_big_no_stream | Librispeech (960小时) | linear | 英文 | ||
deepspeech2 | 超大数据集(1600多小时真实数据)+(1300多小时合成数据) | linear | 中文 | 0.06215 | 点击下载(需要重新导出模型) |
| deepspeech2_big | 超大数据集(1600多小时真实数据)+(1300多小时合成数据) | linear | 中文 | 0.05517 | 先`star`项目再[点击下载](https://pan.baidu.com/s/1IW7HJP16IxRHeqSfMfNK5g?pwd=0w36) |
说明:
- 这里字错率是使用
eval.py
程序并使用集束搜索解码ctc_beam_search
方法计算得到的。 - 中文解码参数为:
alpha=2.2,beta=4.3,beam_size=300,cutoff_prob=0.99,cutoff_top_n=40
。 - 英文解码参数为:
alpha=1.9,beta=0.3,beam_size=500,cutoff_prob=1.0,cutoff_top_n=40
。 - 除了aishell数据集按照数据集本身划分的训练数据和测试数据,其他的都是按照项目设置的固定比例划分训练数据和测试数据。
- 下载的压缩文件已经包含了
mean_std.npz
和vocabulary.txt
,需要把解压得到的全部文件复制到项目根目录下。 - 模型名称包含
no_stream
为非流式模型,不能用于流式识别。
有问题欢迎提 issue 交流
文档教程
快速预测
python infer_path.py --wav_path=./dataset/test.wav
输出结果:
----------- Configuration Arguments -----------
alpha: 1.2
beam_size: 10
beta: 0.35
cutoff_prob: 1.0
cutoff_top_n: 40
decoding_method: ctc_greedy
enable_mkldnn: False
is_long_audio: False
lang_model_path: ./lm/zh_giga.no_cna_cmn.prune01244.klm
mean_std_path: ./dataset/mean_std.npz
model_dir: ./models/infer/
to_an: True
use_gpu: True
use_tensorrt: False
vocab_path: ./dataset/zh_vocab.txt
wav_path: ./dataset/test.wav
------------------------------------------------
消耗时间:132, 识别结果: 近几年不但我用书给女儿儿压岁也劝说亲朋不要给女儿压岁钱而改送压岁书, 得分: 94
- 长语音预测
python infer_path.py --wav_path=./dataset/test_vad.wav --is_long_audio=True
- Web部署
- GUI界面部署
相关项目
- 基于Pytorch实现的声纹识别:VoiceprintRecognition-Pytorch
- 基于Pytorch实现的分类:AudioClassification-Pytorch
- 基于PaddlePaddle实现的语音识别:PPASR
参考资料
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
masr-1.1.0-py3-none-any.whl
(69.0 kB
view hashes)