多语种语音识别模块使用教程¶
一、概述¶
语音识别是一种先进的工具,它能够将人类口述的多种语言自动转换为相应的文本或命令。该技术还在智能客服、语音助手、会议记录等多个领域发挥着重要作用。多语种语音识别则可以支持自动进行语种检索,支持多种不同语言的语音的识别。
二、支持模型列表¶
Whisper Model¶
模型 | 模型下载链接 | 训练数据 | 模型大小 | 词错率 | 介绍 |
---|---|---|---|---|---|
whisper_large | whisper_large | 680kh | 5.8G | 2.7 (Librispeech) | Whisper 是 OpenAI 开发的多语言自动语音识别模型,具备高精度和鲁棒性。它采用端到端架构,能处理嘈杂环境音频,适用于语音助理、实时字幕等多种应用。 |
whisper_medium | whisper_medium | 680kh | 2.9G | - | |
whisper_small | whisper_small | 680kh | 923M | - | |
whisper_base | whisper_base | 680kh | 277M | - | |
whisper_small | whisper_tiny | 680kh | 145M | - |
三、快速集成¶
在快速集成前,首先需要安装 PaddleX 的 wheel 包,wheel的安装方式请参考PaddleX本地安装教程。完成 wheel 包的安装后,几行代码即可完成多语种语音识别模块的推理,可以任意切换该模块下的模型,您也可以将多语种语音识别模块中的模型推理集成到您的项目中。运行以下代码前,请您下载示例语音到本地。
from paddlex import create_model
model = create_model(model_name="whisper_large")
output = model.predict(input="./zh.wav", batch_size=1)
for res in output:
res.print()
res.save_to_json(save_path="./output/res.json")
{'res': {'input_path': './zh.wav', 'result': {'text': '我认为跑步最重要的就是给我带来了身体健康', 'segments': [{'id': 0, 'seek': 0, 'start': 0.0, 'end': 2.0, 'text': '我认为跑步最重要的就是', 'tokens': [50364, 1654, 7422, 97, 13992, 32585, 31429, 8661, 24928, 1546, 5620, 50464, 50464, 49076, 4845, 99, 34912, 19847, 29485, 44201, 6346, 115, 50564], 'temperature': 0, 'avg_logprob': -0.22779104113578796, 'compression_ratio': 0.28169014084507044, 'no_speech_prob': 0.026114309206604958}, {'id': 1, 'seek': 200, 'start': 2.0, 'end': 31.0, 'text': '给我带来了身体健康', 'tokens': [50364, 49076, 4845, 99, 34912, 19847, 29485, 44201, 6346, 115, 51814], 'temperature': 0, 'avg_logprob': -0.21976988017559052, 'compression_ratio': 0.23684210526315788, 'no_speech_prob': 0.009023111313581467}], 'language': 'zh'}}}
input_path
: 输入音频存放路径
- result
: 识别结果
- text
: 语音识别结果文本
- segments
: 带时间戳的结果文本
* id
: ID
* seek
: 语音片段指针
* start
: 片段开始时间
* end
: 片段结束时间
* text
: 片段识别文本
* tokens
: 片段文本的 token id
* temperature
: 变速比例
* avg_logprob
: 平均 log 概率
* compression_ratio
: 压缩比
* no_speech_prob
: 非语音概率
- language
: 识别语种
相关方法、参数等说明如下:
* create_model
多语种识别模型(此处以whisper_large
为例),具体说明如下:
参数 | 参数说明 | 参数类型 | 可选项 | 默认值 |
---|---|---|---|---|
model_name |
模型名称 | str |
whisper_large、whisper_medium、whisper_base、whisper_small、whisper_tiny |
whisper_large |
model_dir |
模型存储路径 | str |
无 | 无 |
-
其中,
model_name
必须指定,指定model_name
后,默认使用 PaddleX 内置的模型参数,在此基础上,指定model_dir
时,使用用户自定义的模型。 -
调用多语种语音识别模型的
predict()
方法进行推理预测,predict()
方法参数有input
和batch_size
,具体说明如下:
参数 | 参数说明 | 参数类型 | 可选项 | 默认值 |
---|---|---|---|---|
input |
待预测数据 | str |
|
无 |
batch_size |
批大小 | int |
目前仅支持1 | 1 |
- 对预测结果进行处理,每个样本的预测结果均为对应的Result对象,支持保存为
json
文件的操作:
方法 | 方法说明 | 参数 | 参数类型 | 参数说明 | 默认值 |
---|---|---|---|---|---|
print() |
打印结果到终端 | format_json |
bool |
是否对输出内容进行使用 JSON 缩进格式化 |
True |
indent |
int |
指定缩进级别,以美化输出的 JSON 数据,使其更具可读性,仅当 format_json 为 True 时有效 |
4 | ||
ensure_ascii |
bool |
控制是否将非 ASCII 字符转义为 Unicode 。设置为 True 时,所有非 ASCII 字符将被转义;False 则保留原始字符,仅当format_json 为True 时有效 |
False |
||
save_to_json() |
将结果保存为json格式的文件 | save_path |
str |
保存的文件路径,当为目录时,保存文件命名与输入文件类型命名一致 | 无 |
indent |
int |
指定缩进级别,以美化输出的 JSON 数据,使其更具可读性,仅当 format_json 为 True 时有效 |
4 | ||
ensure_ascii |
bool |
控制是否将非 ASCII 字符转义为 Unicode 。设置为 True 时,所有非 ASCII 字符将被转义;False 则保留原始字符,仅当format_json 为True 时有效 |
False |
- 此外,也支持通过属性获取预测结果,具体如下:
属性 | 属性说明 |
---|---|
json |
获取预测的json 格式的结果 |
关于更多 PaddleX 的单模型推理的 API 的使用方法,可以参考PaddleX单模型Python脚本使用说明。