pass all parameters explicity to _istft

pull/422/head
Eren Gölge 2021-04-06 11:00:02 +02:00
parent f0e76ee135
commit e0e3b12b26
1 changed files with 14 additions and 9 deletions

View File

@ -306,17 +306,22 @@ class AudioProcessor(object):
### STFT and ISTFT ###
def _stft(self, y):
return librosa.stft(
y=y,
n_fft=self.fft_size,
hop_length=self.hop_length,
win_length=self.win_length,
pad_mode=self.stft_pad_mode,
)
return librosa.stft(y=y,
n_fft=self.fft_size,
hop_length=self.hop_length,
win_length=self.win_length,
pad_mode=self.stft_pad_mode,
window='hann',
center=True,
dtype=None)
def _istft(self, y):
return librosa.istft(
y, hop_length=self.hop_length, win_length=self.win_length)
return librosa.istft(y,
hop_length=self.hop_length,
win_length=self.win_length,
window='hann',
center=True,
dtype=None)
def _griffin_lim(self, S):
angles = np.exp(2j * np.pi * np.random.rand(*S.shape))