VGGishBundle

class torchaudio.prototype.pipelines.VGGishBundle

VGGish [Hershey et al., 2017] 推理流水線，移植自 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