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 backend option for eval

Browse Source
This commit is contained in:
romi1502
2020-07-24 15:01:40 +02:00
parent e8647e3f91
commit b03ef93be9
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
spleeter/commands/__init__.py
View File

@@ -170,6 +170,7 @@ def _create_evaluate_parser(parser_factory):
parser.add_argument('-o', '--output_path', **OPT_OUTPUT) parser.add_argument('-o', '--output_path', **OPT_OUTPUT)
parser.add_argument('--mus_dir', **OPT_MUSDB) parser.add_argument('--mus_dir', **OPT_MUSDB)
parser.add_argument('-m', '--mwf', **OPT_MWF) parser.add_argument('-m', '--mwf', **OPT_MWF)
parser.add_argument('-B', '--stft-backend', **OPT_STFT_BACKEND)
return parser return parser

2
spleeter/commands/evaluate.py
View File

@@ -77,7 +77,7 @@ def _separate_evaluation_dataset(arguments, musdb_root_directory, params):
bitrate='128k', bitrate='128k',
MWF=arguments.MWF, MWF=arguments.MWF,
verbose=arguments.verbose, verbose=arguments.verbose,
stft_backend="auto"), stft_backend=arguments.stft_backend),
params) params)
return audio_output_directory return audio_output_directory