🚀 XLS-R-300M - 布列塔尼语
该模型是 facebook/wav2vec2-xls-r-300m 在 MOZILLA - FOUNDATION/COMMON_VOICE_8_0 - BR 数据集上的微调版本。它在评估集上取得了以下结果:
✨ 主要特性
- 适用于自动语音识别任务。
- 基于 Mozilla 基金会的 Common Voice 8.0 布列塔尼语数据集进行微调。
📦 安装指南
文档中未提及安装步骤,故跳过此章节。
💻 使用示例
基础用法
import torch
from datasets import load_dataset
from transformers import AutoModelForCTC, AutoProcessor
import torchaudio.functional as F
model_id = "infinitejoy/wav2vec2-large-xls-r-300m-breton-cv8"
sample_iter = iter(load_dataset("mozilla-foundation/common_voice_8_0", "br", split="test", streaming=True, use_auth_token=True))
sample = next(sample_iter)
resampled_audio = F.resample(torch.tensor(sample["audio"]["array"]), 48_000, 16_000).numpy()
model = AutoModelForCTC.from_pretrained(model_id)
processor = AutoProcessor.from_pretrained(model_id)
input_values = processor(resampled_audio, return_tensors="pt").input_values
with torch.no_grad():
logits = model(input_values).logits
transcription = processor.batch_decode(logits.numpy()).text
评估命令
1. 在 mozilla-foundation/common_voice_8_0
数据集的 test
分割上进行评估
python eval.py --model_id infinitejoy/wav2vec2-large-xls-r-300m-breton-cv8 --dataset mozilla-foundation/common_voice_8_0 --config br --split test
2. 在 speech-recognition-community-v2/dev_data
数据集上进行评估
python eval.py --model_id infinitejoy/wav2vec2-large-xls-r-300m-breton-cv8 --dataset speech-recognition-community-v2/dev_data --config br --split validation --chunk_length_s 5.0 --stride_length_s 1.0
📚 详细文档
模型信息
评估结果
评估指标 |
数值 |
测试词错误率(Test WER) |
54.855 |
测试字符错误率(Test CER) |
17.865 |
框架版本
- Transformers 4.16.0.dev0
- Pytorch 1.10.0+cu102
- Datasets 1.17.1.dev0
- Tokenizers 0.10.3
📄 许可证
本模型采用 Apache-2.0 许可证。