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

fix: update writing process

Browse Source
This commit is contained in:
Félix Voituret
2019-11-06 18:42:52 +01:00
parent deaadf06a3
commit cce8636fdf
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
spleeter/commands/separate.py
View File

@@ -129,7 +129,6 @@ def process_audio(
yield_single_examples=False) yield_single_examples=False)
# initialize pool for audio export # initialize pool for audio export
pool = Pool(16) pool = Pool(16)
tasks = []
for sample in prediction: for sample in prediction:
sample_filename = sample.pop('audio_id', 'unknown_filename').decode() sample_filename = sample.pop('audio_id', 'unknown_filename').decode()
input_directory, input_filename = split(sample_filename) input_directory, input_filename = split(sample_filename)
@@ -144,13 +143,12 @@ def process_audio(
output_path, output_path,
output_dirname, output_dirname,
f'{instrument}.{codec}') f'{instrument}.{codec}')
tasks.append( pool.apply_async(
pool.apply_async( audio_adapter.save,
audio_adapter.save, (filename, waveform, sample_rate, codec))
(filename, waveform, sample_rate, codec)))
# Wait for everything to be written # Wait for everything to be written
for task in tasks: pool.join()
task.wait(timeout=20) pool.close()
def entrypoint(arguments, params): def entrypoint(arguments, params):

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

@@ -113,7 +113,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
.output(path, format='f32le', **output_kwargs) .output(path, format='f32le', **output_kwargs)
.run_async(pipe_stdin=True, quiet=True)) .run_async(pipe_stdin=True, quiet=True))
try: try:
process.stdin.write(data.astype('<f4').tostring()) process.stdin.write(data.astype('<f4').tobytes())
process.stdin.close() process.stdin.close()
process.wait() process.wait()
except IOError: except IOError: