语言: ca
数据集:
- 通用语音(Common Voice)
- 议会发言(ParlamentParla)
评估指标:
- 词错误率(WER)
标签:
- 音频
- 自动语音识别
- 语音
- 语音转文本
许可证: Apache-2.0
模型索引:
- 名称: 加泰罗尼亚语VoxPopuli Wav2Vec2大型模型
结果:
- 任务:
名称: 语音识别
类型: 自动语音识别
数据集:
- 名称: Common Voice加泰罗尼亚语
类型: common_voice
参数: ca
- 名称: ParlamentParla
URL: https://www.openslr.org/59/
指标:
- 名称: 测试WER
类型: wer
值: 5.98
- 名称: 谷歌众包语料库WER
类型: wer
值: 12.14
- 名称: 有声书《圣乔治传说》WER
类型: wer
值: 12.02
Wav2Vec2-Large-100k-VoxPopuli-加泰罗尼亚语
⚠️注意⚠️: 此模型已迁移至以下URL:
https://huggingface.co/softcatala/wav2vec2-large-100k-voxpopuli-catala
基于facebook/wav2vec2-large-100k-voxpopuli模型,使用Common Voice和ParlamentParla数据集对加泰罗尼亚语进行微调。
注意: 所使用的训练/开发/测试集划分与CommonVoice 6.1数据集不完全对应。采用了结合CommonVoice和ParlamentParla数据集的定制划分方案,详见此处。在CV测试集上评估会产生偏差的WER,因为该数据集中有1144个音频文件被用于本模型的训练/评估过程。
WER是通过此测试集计算的,该测试集在训练/评估过程中未被模型接触。
训练和评估脚本可在GitHub仓库ccoreilly/wav2vec2-catala中找到。
使用此模型时,请确保语音输入采样率为16kHz。
评估结果
词错误率在以下未见数据集上评估得出:
使用方式
该模型可直接使用(无需语言模型)如下:
import torch
import torchaudio
from datasets import load_dataset
from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
test_dataset = load_dataset("common_voice", "ca", split="test[:2%]")
processor = Wav2Vec2Processor.from_pretrained("ccoreilly/wav2vec2-large-100k-voxpopuli-catala")
model = Wav2Vec2ForCTC.from_pretrained("ccoreilly/wav2vec2-large-100k-voxpopuli-catala")
resampler = torchaudio.transforms.Resample(48_000, 16_000)
def speech_file_to_array_fn(batch):
speech_array, sampling_rate = torchaudio.load(batch["path"])
batch["speech"] = resampler(speech_array).squeeze().numpy()
return batch
test_dataset = test_dataset.map(speech_file_to_array_fn)
inputs = processor(test_dataset["speech"][:2], sampling_rate=16_000, return_tensors="pt", padding=True)
with torch.no_grad():
logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits
predicted_ids = torch.argmax(logits, dim=-1)
print("预测结果:", processor.batch_decode(predicted_ids))
print("参考文本:", test_dataset["sentence"][:2])