• 文档 >
  • torchaudio.pipelines >
  • 旧版本 (稳定版)
快捷方式

torchaudio.pipelines

torchaudio.pipelines 模块将预训练模型与支持函数和元数据打包到简单的 API 中,这些 API 专为执行特定任务而设计。

当使用预训练模型执行任务时,除了使用预训练权重实例化模型之外,客户端代码还需要以与训练期间相同的方式构建特征提取和后处理管道。这需要将训练期间使用到的信息传递过来,例如变换类型及其参数(例如,采样率和 FFT 窗口大小)。

为了将这些信息与预训练模型关联并易于访问,torchaudio.pipelines 模块使用了 Bundle 类的概念,该类定义了一组用于实例化管道以及管道接口的 API。

下图说明了这一点。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-intro.png

预训练模型和相关管道表示为 Bundle 的一个实例。相同 Bundle 的不同实例共享相同的接口,但它们的实现不受类型限制。例如,SourceSeparationBundle 定义了执行源分离的接口,但其实例 CONVTASNET_BASE_LIBRI2MIX 实例化了一个 ConvTasNet 模型,而 HDEMUCS_HIGH_MUSDB 实例化了一个 HDemucs 模型。尽管如此,由于它们共享相同的接口,因此用法相同。

注意

在幕后,Bundle 的实现使用了来自其他 torchaudio 模块的组件,例如 torchaudio.modelstorchaudio.transforms,甚至像 SentencPieceDeepPhonemizer 这样的第三方库。但是,此实现细节对库用户来说是抽象的。

RNN-T 流式/非流式 ASR

接口

RNNTBundle 定义了 ASR 管道,包含三个步骤:特征提取、推理和去标记化。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-rnntbundle.png

RNNTBundle

一个数据类,捆绑了使用 RNN-T 模型执行自动语音识别 (ASR,语音到文本) 推理的组件。

RNNTBundle.FeatureExtractor

RNN-T 管道中特征提取部分的接口

RNNTBundle.TokenProcessor

RNN-T 管道中标记处理器部分的接口

使用 RNNTBundle 的教程

Online ASR with Emformer RNN-T

使用 Emformer RNN-T 进行在线 ASR

使用 Emformer RNN-T 进行在线 ASR
Device ASR with Emformer RNN-T

使用 Emformer RNN-T 进行设备端 ASR

使用 Emformer RNN-T 进行设备端 ASR

预训练模型

EMFORMER_RNNT_BASE_LIBRISPEECH

基于 Emformer-RNNT 的 ASR 管道,在 LibriSpeech 数据集 [Panayotov 等人,2015] 上进行预训练,能够执行流式和非流式推理。

wav2vec 2.0 / HuBERT / WavLM - SSL

接口

Wav2Vec2Bundle 实例化生成声学特征的模型,这些特征可用于下游推理和微调。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-wav2vec2bundle.png

Wav2Vec2Bundle

一个数据类,捆绑了使用预训练 Wav2Vec2Model 的相关信息。

预训练模型

WAV2VEC2_BASE

Wav2vec 2.0 模型(“base” 架构),在来自 LibriSpeech 数据集 [Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和“train-other-500” 的组合)的 960 小时未标记音频上进行预训练,未进行微调。

WAV2VEC2_LARGE

Wav2vec 2.0 模型(“large” 架构),在来自 LibriSpeech 数据集 [Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和“train-other-500” 的组合)的 960 小时未标记音频上进行预训练,未进行微调。

WAV2VEC2_LARGE_LV60K

Wav2vec 2.0 模型(“large-lv60k” 架构),在来自 Libri-Light 数据集 [Kahn 等人,2020] 的 60,000 小时未标记音频上进行预训练,未进行微调。

WAV2VEC2_XLSR53

Wav2vec 2.0 模型(“base” 架构),在来自多个数据集的 56,000 小时未标注音频上进行预训练(Multilingual LibriSpeech [Pratap 等人,2020]CommonVoice [Ardila 等人,2020]BABEL [Gales 等人,2014]),未进行微调。

WAV2VEC2_XLSR_300M

XLS-R 模型,拥有 3 亿个参数,在来自多个数据集的 436,000 小时未标注音频上进行预训练(Multilingual LibriSpeech [Pratap 等人,2020]CommonVoice [Ardila 等人,2020]VoxLingua107 [Valk 和 Alumäe,2021]BABEL [Gales 等人,2014]VoxPopuli [Wang 等人,2021])涵盖 128 种语言,未进行微调。

WAV2VEC2_XLSR_1B

XLS-R 模型,拥有 10 亿个参数,在来自多个数据集的 436,000 小时未标注音频上进行预训练(Multilingual LibriSpeech [Pratap 等人,2020]CommonVoice [Ardila 等人,2020]VoxLingua107 [Valk 和 Alumäe,2021]BABEL [Gales 等人,2014]VoxPopuli [Wang 等人,2021])涵盖 128 种语言,未进行微调。

WAV2VEC2_XLSR_2B

XLS-R 模型,拥有 20 亿个参数,在来自多个数据集的 436,000 小时未标注音频上进行预训练(Multilingual LibriSpeech [Pratap 等人,2020]CommonVoice [Ardila 等人,2020]VoxLingua107 [Valk 和 Alumäe,2021]BABEL [Gales 等人,2014]VoxPopuli [Wang 等人,2021])涵盖 128 种语言,未进行微调。

HUBERT_BASE

HuBERT 模型(“base” 架构),在 LibriSpeech 数据集 [Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标注音频上进行预训练,未进行微调。

HUBERT_LARGE

HuBERT 模型(“large” 架构),在 Libri-Light 数据集 [Kahn 等人,2020]的 60,000 小时未标注音频上进行预训练,未进行微调。

HUBERT_XLARGE

HuBERT 模型(“extra large” 架构),在 Libri-Light 数据集 [Kahn 等人,2020]的 60,000 小时未标注音频上进行预训练,未进行微调。

WAVLM_BASE

WavLM Base 模型(“base” 架构),在 LibriSpeech 数据集 [Panayotov 等人,2015]的 960 小时未标注音频上进行预训练,未进行微调。

WAVLM_BASE_PLUS

WavLM Base+ 模型(“base” 架构),在 Libri-Light 数据集 [Kahn 等人,2020]的 60,000 小时、GigaSpeech [Chen 等人,2021] 的 10,000 小时和 VoxPopuli [Wang 等人,2021] 的 24,000 小时未标注音频上进行预训练,未进行微调。

WAVLM_LARGE

WavLM Large 模型(“large” 架构),在 Libri-Light 数据集 [Kahn 等人,2020]的 60,000 小时、GigaSpeech [Chen 等人,2021] 的 10,000 小时和 VoxPopuli [Wang 等人,2021] 的 24,000 小时未标注音频上进行预训练,未进行微调。

wav2vec 2.0 / HuBERT - 微调的 ASR

接口

Wav2Vec2ASRBundle 实例化生成预定义标签概率分布的模型,可用于 ASR。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-wav2vec2asrbundle.png

Wav2Vec2ASRBundle

一个数据类,捆绑了使用预训练 Wav2Vec2Model 的相关信息。

使用 Wav2Vec2ASRBundle 的教程

Speech Recognition with Wav2Vec2

使用 Wav2Vec2 进行语音识别

使用 Wav2Vec2 进行语音识别
ASR Inference with CTC Decoder

使用 CTC 解码器进行 ASR 推理

使用 CTC 解码器进行 ASR 推理
Forced Alignment with Wav2Vec2

使用 Wav2Vec2 进行强制对齐

使用 Wav2Vec2 进行强制对齐

预训练模型

WAV2VEC2_ASR_BASE_10M

Wav2vec 2.0 模型(“base” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在来自 Libri-Light 数据集[Kahn 等人,2020](“train-10min” 子集)的 10 分钟转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_BASE_100H

Wav2vec 2.0 模型(“base” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在 “train-clean-100” 子集的 100 小时转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_BASE_960H

Wav2vec 2.0 模型(“base” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在相同音频及其对应的转录文本上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_10M

Wav2vec 2.0 模型(“large” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在来自 Libri-Light 数据集[Kahn 等人,2020](“train-10min” 子集)的 10 分钟转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_100H

Wav2vec 2.0 模型(“large” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在相同数据集(“train-clean-100” 子集)的 100 小时转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_960H

Wav2vec 2.0 模型(“large” 架构,带有一个额外的线性模块),在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时未标记音频上进行预训练,并在相同音频及其对应的转录文本上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_LV60K_10M

Wav2vec 2.0 模型(“large-lv60k” 架构,带有一个额外的线性模块),在来自 Libri-Light 数据集[Kahn 等人,2020] 的 60,000 小时未标记音频上进行预训练,并在相同数据集(“train-10min” 子集)的 10 分钟转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_LV60K_100H

Wav2vec 2.0 模型(“large-lv60k” 架构,带有一个额外的线性模块),在来自 Libri-Light 数据集[Kahn 等人,2020] 的 60,000 小时未标记音频上进行预训练,并在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100” 子集)的 100 小时转录音频上针对 ASR 进行微调。

WAV2VEC2_ASR_LARGE_LV60K_960H

Wav2vec 2.0 模型(“large-lv60k” 架构,带有一个额外的线性模块),在来自 Libri-Light [Kahn 等人,2020]数据集的 60,000 小时未标记音频上进行预训练,并在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时转录音频上针对 ASR 进行微调。

VOXPOPULI_ASR_BASE_10K_DE

wav2vec 2.0 模型(“base” 架构),在来自 VoxPopuli 数据集[Wang 等人,2021](“10k” 子集,包含 23 种语言)的 10k 小时未标记音频上进行预训练,并在 “de” 子集的 282 小时转录音频上针对 ASR 进行微调。

VOXPOPULI_ASR_BASE_10K_EN

wav2vec 2.0 模型(“base” 架构),在来自 VoxPopuli 数据集[Wang 等人,2021](“10k” 子集,包含 23 种语言)的 10k 小时未标记音频上进行预训练,并在 “en” 子集的 543 小时转录音频上针对 ASR 进行微调。

VOXPOPULI_ASR_BASE_10K_ES

wav2vec 2.0 模型(“base” 架构),在来自 VoxPopuli 数据集[Wang 等人,2021](“10k” 子集,包含 23 种语言)的 10k 小时未标记音频上进行预训练,并在 “es” 子集的 166 小时转录音频上针对 ASR 进行微调。

VOXPOPULI_ASR_BASE_10K_FR

wav2vec 2.0 模型(“base” 架构),在来自 VoxPopuli 数据集[Wang 等人,2021](“10k” 子集,包含 23 种语言)的 10k 小时未标记音频上进行预训练,并在 “fr” 子集的 211 小时转录音频上针对 ASR 进行微调。

VOXPOPULI_ASR_BASE_10K_IT

wav2vec 2.0 模型(“base” 架构),在来自 VoxPopuli 数据集[Wang 等人,2021](“10k” 子集,包含 23 种语言)的 10k 小时未标记音频上进行预训练,并在 “it” 子集的 91 小时转录音频上针对 ASR 进行微调。

HUBERT_ASR_LARGE

HuBERT 模型(“large” 架构),在来自 Libri-Light 数据集[Kahn 等人,2020] 的 60,000 小时未标记音频上进行预训练,并在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360” 和 “train-other-500” 的组合)的 960 小时转录音频上针对 ASR 进行微调。

HUBERT_ASR_XLARGE

HuBERT 模型(“超大”架构),在来自 Libri-Light 数据集[Kahn 等人,2020] 的 60,000 小时未标记音频上进行预训练,并在来自 LibriSpeech 数据集[Panayotov 等人,2015](“train-clean-100”、“train-clean-360”和“train-other-500”的组合)的 960 小时转录音频上针对 ASR 进行微调。

wav2vec 2.0 / HuBERT - 强制对齐

接口

Wav2Vec2FABundle 将预训练模型及其关联字典捆绑在一起。此外,它支持追加 star 令牌维度。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-wav2vec2fabundle.png

Wav2Vec2FABundle

数据类,捆绑相关信息以使用预训练的 Wav2Vec2Model 进行强制对齐。

Wav2Vec2FABundle.Tokenizer

分词器的接口

Wav2Vec2FABundle.Aligner

对齐器的接口

使用 Wav2Vec2FABundle 的教程

CTC forced alignment API tutorial

CTC 强制对齐 API 教程

CTC 强制对齐 API 教程
Forced alignment for multilingual data

多语言数据的强制对齐

多语言数据的强制对齐
Forced Alignment with Wav2Vec2

使用 Wav2Vec2 进行强制对齐

使用 Wav2Vec2 进行强制对齐

预训练模型

MMS_FA

在来自 Scaling Speech Technology to 1,000+ Languages [Pratap 等人,2023] 的 1,130 种语言的 31K 小时数据上训练。

Tacotron2 文本转语音

Tacotron2TTSBundle 定义了文本转语音管道,包括三个步骤:分词、频谱图生成和声码器。频谱图生成基于 Tacotron2 模型。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-tacotron2bundle.png

TextProcessor 可以是基于规则的分词(字符的情况下),也可以是基于神经网络的 G2P 模型,该模型从输入文本生成音素序列。

类似地,Vocoder 可以是不带学习参数的算法,如 Griffin-Lim,也可以是基于神经网络的模型,如 Waveglow

接口

Tacotron2TTSBundle

数据类,捆绑相关信息以使用预训练的 Tacotron2 和声码器。

Tacotron2TTSBundle.TextProcessor

Tacotron2TTS 管道文本处理部分的接口

Tacotron2TTSBundle.Vocoder

Tacotron2TTS 管道声码器部分的接口

使用 Tacotron2TTSBundle 的教程

Text-to-Speech with Tacotron2

使用 Tacotron2 进行文本转语音

使用 Tacotron2 进行文本转语音

预训练模型

TACOTRON2_WAVERNN_PHONE_LJSPEECH

基于音素的 TTS 管道,使用在 LJSpeech [Ito 和 Johnson,2017] 上训练了 1,500 个 epoch 的 Tacotron2,以及在 LJSpeech [Ito 和 Johnson,2017] 的 8 位深度波形上训练了 10,000 个 epoch 的 WaveRNN 声码器。

TACOTRON2_WAVERNN_CHAR_LJSPEECH

基于字符的 TTS 管道,使用在 LJSpeech [Ito 和 Johnson,2017] 上训练了 1,500 个 epoch 的 Tacotron2,以及在 LJSpeech [Ito 和 Johnson,2017] 的 8 位深度波形上训练了 10,000 个 epoch 的 WaveRNN 声码器。

TACOTRON2_GRIFFINLIM_PHONE_LJSPEECH

基于音素的 TTS 管道,使用在 LJSpeech [Ito 和 Johnson,2017] 上训练了 1,500 个 epoch 的 Tacotron2,以及 GriffinLim 作为声码器。

TACOTRON2_GRIFFINLIM_CHAR_LJSPEECH

基于字符的 TTS 管道,使用在 LJSpeech [Ito 和 Johnson,2017] 上训练了 1,500 个 epoch 的 Tacotron2,以及 GriffinLim 作为声码器。

声源分离

接口

SourceSeparationBundle 实例化声源分离模型,这些模型接收单通道音频并生成多通道音频。

https://download.pytorch.org/torchaudio/doc-assets/pipelines-sourceseparationbundle.png

SourceSeparationBundle

数据类,捆绑用于执行声源分离的组件。

使用 SourceSeparationBundle 的教程

Music Source Separation with Hybrid Demucs

使用 Hybrid Demucs 进行音乐源分离

使用 Hybrid Demucs 进行音乐源分离

预训练模型

CONVTASNET_BASE_LIBRI2MIX

预训练的声源分离管道,使用在 Libri2Mix 数据集 [Cosentino 等人,2020] 上训练的 ConvTasNet [Luo 和 Mesgarani,2019]

HDEMUCS_HIGH_MUSDB_PLUS

预训练的音乐声源分离管道,使用在 MUSDB-HQ [Rafii 等人,2019] 的训练集和测试集以及 Meta 内部数据库中额外 150 首专门制作的歌曲上训练的 Hybrid Demucs [Défossez,2021]

HDEMUCS_HIGH_MUSDB

预训练的音乐声源分离管道,使用在 MUSDB-HQ [Rafii 等人,2019] 的训练集上训练的 Hybrid Demucs [Défossez,2021]

Squim 目标

接口

SquimObjectiveBundle 定义了语音质量和可懂度测量 (SQUIM) 管道,该管道可以根据输入波形预测**客观**指标得分。

SquimObjectiveBundle

数据类,捆绑相关信息以使用预训练的 SquimObjective 模型。

预训练模型

SQUIM_OBJECTIVE

使用 [Kumar 等人,2023] 中描述的方法在 DNS 2020 数据集 [Reddy 等人,2020] 上训练的 SquimObjective 管道。

Squim 主观

接口

SquimSubjectiveBundle 定义了语音质量和可懂度测量 (SQUIM) 管道,该管道可以根据输入波形预测**主观**指标得分。

SquimSubjectiveBundle

用于捆绑相关信息以使用预训练的SquimSubjective 模型的数据类。

预训练模型

SQUIM_SUBJECTIVE

SquimSubjective 管道根据[Manocha 和 Kumar,2022][Kumar 等人,2023] 中描述的方法,在 BVCC [Cooper 和 Yamagishi,2021]DAPS [Mysore,2014] 数据集上进行了训练。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源