Class: TorchAudio::Transforms::MelSpectrogram

Inherits:
Torch::NN::Module
  • Object
show all
Defined in:
lib/torchaudio/transforms/mel_spectrogram.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(sample_rate: 16000, n_fft: 400, win_length: nil, hop_length: nil, f_min: 0.0, f_max: nil, pad: 0, n_mels: 128, window_fn: Torch.method(:hann_window), power: 2.0, normalized: false, wkwargs: nil) ⇒ MelSpectrogram

Returns a new instance of MelSpectrogram.



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/torchaudio/transforms/mel_spectrogram.rb', line 6

def initialize(
  sample_rate: 16000, n_fft: 400, win_length: nil, hop_length: nil, f_min: 0.0,
  f_max: nil, pad: 0, n_mels: 128, window_fn: Torch.method(:hann_window),
  power: 2.0, normalized: false, wkwargs: nil
)

  super()
  @sample_rate = sample_rate
  @n_fft = n_fft
  @win_length = win_length || n_fft
  @hop_length = hop_length || @win_length.div(2)
  @pad = pad
  @power = power
  @normalized = normalized
  @n_mels = n_mels  # number of mel frequency bins
  @f_max = f_max
  @f_min = f_min
  @spectrogram =
    Spectrogram.new(
      n_fft: @n_fft, win_length: @win_length, hop_length: @hop_length, pad: @pad,
      window_fn: window_fn, power: @power, normalized: @normalized, wkwargs: wkwargs
    )
  @mel_scale = MelScale.new(n_mels: @n_mels, sample_rate: @sample_rate, f_min: @f_min, f_max: @f_max, n_stft: @n_fft.div(2) + 1)
end

Instance Attribute Details

#n_mels ⇒ Object (readonly)

Returns the value of attribute n_mels.



4
5
6
 
# File 'lib/torchaudio/transforms/mel_spectrogram.rb', line 4

def n_mels
  @n_mels
end

Instance Method Details

#forward(waveform) ⇒ Object 



31
32
33
34
 
# File 'lib/torchaudio/transforms/mel_spectrogram.rb', line 31

def forward(waveform)
  specgram = @spectrogram.call(waveform)
  @mel_scale.call(specgram)
end