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

Added padding at the begining to avoid tf STFT reconstruction erro

Browse Source
This commit is contained in:
romi1502
2020-07-03 17:25:58 +02:00
parent 39af9502ab
commit e8647e3f91
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
spleeter/model/__init__.py
View File

@@ -275,9 +275,16 @@ class EstimatorSpecBuilder(object):
spec_name = self.spectrogram_name spec_name = self.spectrogram_name
if stft_name not in self._features: if stft_name not in self._features:
# pad input with a frame of zeros
waveform = tf.concat([
tf.zeros((self._frame_length, self._n_channels)),
self._features['waveform']
],
0
)
stft_feature = tf.transpose( stft_feature = tf.transpose(
stft( stft(
tf.transpose(self._features['waveform']), tf.transpose(waveform),
self._frame_length, self._frame_length,
self._frame_step, self._frame_step,
window_fn=lambda frame_length, dtype: ( window_fn=lambda frame_length, dtype: (
@@ -341,7 +348,7 @@ class EstimatorSpecBuilder(object):
reshaped = tf.transpose(inversed) reshaped = tf.transpose(inversed)
if time_crop is None: if time_crop is None:
time_crop = tf.shape(self._features['waveform'])[0] time_crop = tf.shape(self._features['waveform'])[0]
return reshaped[:time_crop, :] return reshaped[self._frame_length:self._frame_length+time_crop, :]
def _build_mwf_output_waveform(self): def _build_mwf_output_waveform(self):
""" Perform separation with multichannel Wiener Filtering using Norbert. """ Perform separation with multichannel Wiener Filtering using Norbert.