| | |
| | |
| | """ |
| | @File : spec.py |
| | @Time : 2023/8/8 下午5:10 |
| | @Author : waytan |
| | @Contact : waytan@tencent.com |
| | @License : (C)Copyright 2023, Tencent |
| | @Desc : Spec |
| | """ |
| |
|
| | import torch as th |
| |
|
| |
|
| | def spectro(x, n_fft=512, hop_length=None, pad=0): |
| | *other, length = x.shape |
| | x = x.reshape(-1, length) |
| | is_mps = x.device.type == 'mps' |
| | if is_mps: |
| | x = x.cpu() |
| | 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) |
| | is_mps = z.device.type == 'mps' |
| | if is_mps: |
| | z = z.cpu() |
| | 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) |
| |
|
