Hub documentation

在 Hugging Face 上使用 SpeechBrain

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

在 Hugging Face 上使用 SpeechBrain

speechbrain 是一个开源的一体化音频/语音对话工具包。其目标是创建一个单一、灵活且用户友好的工具包,以便轻松开发最先进的语音技术,包括语音识别、说话人识别、语音增强、语音分离、语言识别、多麦克风信号处理等众多系统。

在 Hub 中探索 SpeechBrain

您可以在模型页面左侧进行筛选,以查找speechbrain模型。

Hub 上的所有模型都具备以下功能:

  1. 自动生成的模型卡片,包含简要描述。
  2. 元数据标签,包含语言、许可证、论文等信息,有助于提高可发现性。
  3. 一个交互式小部件,可直接在浏览器中用于体验该模型。
  4. 一个用于发起推理请求的 Inference Providers 小部件。

使用现有模型

speechbrain 提供了不同的接口来管理针对不同任务的预训练模型,例如 EncoderClassifier、EncoderClassifier、SepformerSeparation 和 SpectralMaskEnhancement。这些类都有一个 from_hparams 方法,你可以使用它从 Hub 加载模型

以下是一个在城市声音中进行声音识别推理的示例。

import torchaudio
from speechbrain.pretrained import EncoderClassifier

classifier = EncoderClassifier.from_hparams(
    source="speechbrain/urbansound8k_ecapa"
)
out_prob, score, index, text_lab = classifier.classify_file('speechbrain/urbansound8k_ecapa/dog_bark.wav')

如果你想了解如何加载特定模型,可以单击 Use in speechbrain,你将获得一个可用的代码片段来加载该模型!

其他资源

Update on GitHub