File size: 1,465 Bytes
1bc2162 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | #!/usr/bin/env python3
# coding: utf-8
# @Author : Xinhao Mei @CVSSP, University of Surrey
# @E-mail : x.mei@surrey.ac.uk
import torch
import torch.nn as nn
from torchlibrosa import LogmelFilterBank, Spectrogram
class AudioFeature(nn.Module):
def __init__(self, audio_config):
super().__init__()
self.mel_trans = Spectrogram(n_fft=audio_config["n_fft"],
hop_length=audio_config["hop_length"],
win_length=audio_config["n_fft"],
window='hann',
center=True,
pad_mode='reflect',
freeze_parameters=True)
self.log_trans = LogmelFilterBank(sr=audio_config["sr"],
n_fft=audio_config["n_fft"],
n_mels=audio_config["n_mels"],
fmin=audio_config["f_min"],
fmax=audio_config["f_max"],
ref=1.0,
amin=1e-10,
top_db=None,
freeze_parameters=True)
def forward(self, input):
mel_feats = self.mel_trans(input)
log_mel = self.log_trans(mel_feats)
return log_mel
|