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

fix: load method

Browse Source
This commit is contained in:
Félix Voituret
2019-11-06 18:18:30 +01:00
parent c23270a5df
commit 7f759e0695
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
spleeter/utils/audio/ffmpeg.py
View File

@@ -60,7 +60,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
raise IOError('No stream was found with ffprobe')
metadata = next(
stream
for stream in probe['stream']
for stream in probe['streams']
if stream['codec_type'] == 'audio')
n_channels = metadata['channels']
if sample_rate is None:
@@ -75,7 +75,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
.input(path, **input_kwargs)
.output('-', format='f32le')
.overwrite_output()
.run(quiet=True))
.run(capture_stdout=True, capture_stderr=True))
waveform = np.frombuffer(buffer, dtype='<f4').reshape(-1, n_channels)
if not waveform.dtype == np.dtype(dtype):
waveform = waveform.astype(dtype)
@@ -112,8 +112,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
.output(path, format='f32le', **output_kwargs)
.run_async(pipe_stdin=True))
try:
process.stdin.write(
data.astype('<f4').tostring())
process.stdin.write(data.astype('<f4').tostring())
process.stdin.close()
process.wait()
except IOError: