VGGishBundle

class torchaudio.prototype.pipelines.VGGishBundle

VGGish [Hershey et al., 2017] 推理 pipeline 从 torchvggish 和 tensorflow-models 移植而来。

示例

>>> import torchaudio
>>> from torchaudio.prototype.pipelines import VGGISH
>>>
>>> input_sr = VGGISH.sample_rate
>>> input_proc = VGGISH.get_input_processor()
>>> model = VGGISH.get_model()
>>>
>>> waveform, sr = torchaudio.load(
>>>     "Chopin_Ballade_-1_In_G_Minor,_Op._23.mp3",
>>> )
>>> waveform = waveform.squeeze(0)
>>> waveform = torchaudio.functional.resample(waveform, sr, input_sr)
>>> mono_output = model(input_proc(waveform))

属性

sample_rate

property VGGishBundle.sample_rate: int

输入处理器和模型期望的输入波形的采样率。

类型:

int (整数)

方法

get_input_processor

VGGishBundle.get_input_processor() → VGGishInputProcessor

构建 VGGish 的输入处理器。

返回:

VGGish 的输入处理器。

返回类型:

VGGishInputProcessor

get_model

VGGishBundle.get_model() → VGGish

构建预训练的 VGGish 模型。必要时下载并缓存权重。

返回:

加载了预训练权重的 VGGish 模型。

返回类型:

VGGish