MelScale

class torchaudio.transforms.MelScale(n_mels: int = 128, sample_rate: int = 16000, f_min: float = 0.0, f_max: Optional[float] = None, n_stft: int = 201, norm: Optional[str] = None, mel_scale: str = 'htk')

将普通的 STFT 转换为具有三角形滤波器组的梅尔频率 STFT。

参数

• n_mels (int, 可选) – 梅尔滤波器组的数量。(默认：128)

• sample_rate (int, 可选) – 音频信号的采样率。(默认：16000)

• f_min (float, 可选) – 最小频率。(默认：0.)

• f_max (float 或 None, 可选) – 最大频率。(默认：sample_rate // 2)

• n_stft (int, 可选) – STFT 中的 bin 数量。请参阅 Spectrogram 中的 n_fft。(默认：201)

• norm (str 或 None, 可选) – 如果为 "slaney"，则将三角形梅尔权重除以梅尔带的宽度（面积归一化）。(默认：None)

• mel_scale (str, 可选) – 使用的比例：htk 或 slaney。(默认：htk)

示例

>>> waveform, sample_rate = torchaudio.load("test.wav", normalize=True)
>>> spectrogram_transform = transforms.Spectrogram(n_fft=1024)
>>> spectrogram = spectrogram_transform(waveform)
>>> melscale_transform = transforms.MelScale(sample_rate=sample_rate, n_stft=1024 // 2 + 1)
>>> melscale_spectrogram = melscale_transform(spectrogram)

另请参阅

torchaudio.functional.melscale_fbanks() - 用于生成滤波器组的函数。

forward(specgram: Tensor) → Tensor

参数

specgram (Tensor) – 维度为 (..., freq, time) 的 STFT 频谱图。

返回

梅尔频率频谱图，大小为 (..., n_mels, time)。

返回类型

张量