From b03ef93be974bfb702af2b10fe9ffd1dd6704974 Mon Sep 17 00:00:00 2001 From: romi1502 Date: Fri, 24 Jul 2020 15:01:40 +0200 Subject: [PATCH] Added backend option for eval --- spleeter/commands/__init__.py | 1 + spleeter/commands/evaluate.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/spleeter/commands/__init__.py b/spleeter/commands/__init__.py index 995e866..ae5f184 100644 --- a/spleeter/commands/__init__.py +++ b/spleeter/commands/__init__.py @@ -170,6 +170,7 @@ def _create_evaluate_parser(parser_factory): parser.add_argument('-o', '--output_path', **OPT_OUTPUT) parser.add_argument('--mus_dir', **OPT_MUSDB) parser.add_argument('-m', '--mwf', **OPT_MWF) + parser.add_argument('-B', '--stft-backend', **OPT_STFT_BACKEND) return parser diff --git a/spleeter/commands/evaluate.py b/spleeter/commands/evaluate.py index 1e1e3f1..35c7181 100644 --- a/spleeter/commands/evaluate.py +++ b/spleeter/commands/evaluate.py @@ -77,7 +77,7 @@ def _separate_evaluation_dataset(arguments, musdb_root_directory, params): bitrate='128k', MWF=arguments.MWF, verbose=arguments.verbose, - stft_backend="auto"), + stft_backend=arguments.stft_backend), params) return audio_output_directory