语言: 葡萄牙语
数据集:

• Common Voice
  评估指标:
• 词错误率 (wer)
  标签:
• 音频
• 语音
• wav2vec2
• 葡萄牙语
• 葡萄牙语语音语料库
• 自动语音识别
• 语音
• PyTorch
  许可证: Apache-2.0
  模型索引:
• 名称: Edresson Casanova Wav2vec2 Large 100k Voxpopuli 使用 Common Voice 7.0、TTS-葡萄牙语语料库及数据增强方法在葡萄牙语上微调
  结果:
  • 任务:
    名称: 语音识别
    类型: 自动语音识别
    指标:
    • 名称: Common Voice 7.0 测试集词错误率
      类型: wer
      值: 20.20

使用 Common Voice 7.0、TTS-葡萄牙语语料库及数据增强方法微调的 Wav2vec2 Large 100k Voxpopuli 葡萄牙语模型

Wav2vec2 Large 100k Voxpopuli 模型基于 TTS 和语音转换的数据增强方法，在 Common Voice 7.0 和 TTS-葡萄牙语数据集上进行了葡萄牙语微调。

使用该模型

from transformers import AutoTokenizer, Wav2Vec2ForCTC  

tokenizer = AutoTokenizer.from_pretrained("Edresson/wav2vec2-large-100k-voxpopuli-ft-Common_Voice_plus_TTS-Dataset_plus_Data_Augmentation-portuguese")  

model = Wav2Vec2ForCTC.from_pretrained("Edresson/wav2vec2-large-100k-voxpopuli-ft-Common_Voice_plus_TTS-Dataset_plus_Data_Augmentation-portuguese")  

结果

详细结果请查阅论文

使用 Common Voice 数据集测试示例

dataset = load_dataset("common_voice", "ru", split="test", data_dir="./cv-corpus-7.0-2021-07-21")  

resampler = torchaudio.transforms.Resample(orig_freq=48_000, new_freq=16_000)  

def map_to_array(batch):  
    speech, _ = torchaudio.load(batch["path"])  
    batch["speech"] = resampler.forward(speech.squeeze(0)).numpy()  
    batch["sampling_rate"] = resampler.new_freq  
    batch["sentence"] = re.sub(chars_to_ignore_regex, '', batch["sentence"]).lower().replace("’", "'")  
    return batch  

ds = dataset.map(map_to_array)  
result = ds.map(map_to_pred, batched=True, batch_size=1, remove_columns=list(ds.features.keys()))  
print(wer.compute(predictions=result["predicted"], references=result["target"]))