yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing gltches issues with Istft

Browse Source
This commit is contained in:
mmoussallam
2020-06-18 18:01:03 +02:00
parent ca5cdd7d28
commit 27f52b71d5
3 changed files with 56 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spleeter/separator.py
View File

@@ -122,16 +122,20 @@ class Separator(object):
assert not (inverse and length is None)
data = np.asfortranarray(data)
N = self._params["frame_length"]
pad_edges = int(N/4)
H = self._params["frame_step"]
win = hann(N, sym=False)
fstft = istft if inverse else stft
win_len_arg = {"win_length": None, "length": length} if inverse else {"n_fft": N}
win_len_arg = {"win_length": None, "length": length + 2*pad_edges} if inverse else {"n_fft": N}
n_channels = data.shape[-1]
out = []
for c in range(n_channels):
d = data[:, :, c].T if inverse else data[:, c]
for c in range(n_channels):
d = data[:, :, c].T if inverse else np.concatenate((np.zeros(pad_edges,), data[:,c], np.zeros(pad_edges,)))
s = fstft(d, hop_length=H, window=win, center=False, **win_len_arg)
if inverse:
s = s[pad_edges:-pad_edges]
s = np.expand_dims(s.T, 2-inverse)
out.append(s)
if len(out) == 1:
return out[0]