| |
| |
| |
| |
| |
| """Conveniance wrapper to perform STFT and iSTFT""" |
|
|
| import torch as th |
|
|
|
|
| def spectro(x, n_fft=512, hop_length=None, pad=0): |
| *other, length = x.shape |
| x = x.reshape(-1, length) |
| z = th.stft(x, |
| n_fft * (1 + pad), |
| hop_length or n_fft // 4, |
| window=th.hann_window(n_fft).to(x), |
| win_length=n_fft, |
| normalized=True, |
| center=True, |
| return_complex=True, |
| pad_mode='reflect') |
| _, freqs, frame = z.shape |
| return z.view(*other, freqs, frame) |
|
|
|
|
| def ispectro(z, hop_length=None, length=None, pad=0): |
| *other, freqs, frames = z.shape |
| n_fft = 2 * freqs - 2 |
| z = z.view(-1, freqs, frames) |
| win_length = n_fft // (1 + pad) |
| x = th.istft(z, |
| n_fft, |
| hop_length, |
| window=th.hann_window(win_length).to(z.real), |
| win_length=win_length, |
| normalized=True, |
| length=length, |
| center=True) |
| _, length = x.shape |
| return x.view(*other, length) |
|
|
