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

Hack so that multichannel doesn't crash with librosa

Browse Source
This commit is contained in:
akhlif
2020-03-27 16:15:14 +01:00
parent 38bfff833e
commit f8c728a67b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
spleeter/separator.py
View File

@@ -145,8 +145,10 @@ class Separator(object):
# TODO: fix the logic, build sometimes return, sometimes set attribute # TODO: fix the logic, build sometimes return, sometimes set attribute
outputs = builder.outputs outputs = builder.outputs
stft = self.stft(waveform) stft = self.stft(waveform)
if stft.shape[-1] != 2: if stft.shape[-1] == 1:
stft = np.concatenate([stft, stft], axis=-1) stft = np.concatenate([stft, stft], axis=-1)
elif stft.shape[-1] > 2:
stft = stft[:, :2]
saver = tf.train.Saver() saver = tf.train.Saver()
with tf.Session() as sess: with tf.Session() as sess: