🐛 fix evaluation test

This commit is contained in:
Faylixe
2020-12-08 14:22:31 +01:00
parent c1e52f2960
commit 8a0e420080
4 changed files with 14 additions and 9 deletions

View File

@@ -189,15 +189,18 @@ def evaluate(
mixtures = [join(song, EVALUATION_MIXTURE) for song in songs]
audio_output_directory = join(output_path, EVALUATION_AUDIO_DIRECTORY)
separate(
adapter=adapter,
params_filename=params_filename,
files=mixtures,
output_path=output_path,
filename_format='{foldername}/{instrument}.{codec}',
adapter=adapter,
bitrate='128k',
codec=Codec.WAV,
duration=600.,
offset=0,
output_path=output_path,
stft_backend=stft_backend,
filename_format='{foldername}/{instrument}.{codec}',
params_filename=params_filename,
mwf=mwf,
verbose=verbose,
stft_backend=stft_backend)
verbose=verbose,)
# Compute metrics with musdb.
metrics_output_directory = join(output_path, EVALUATION_METRICS_DIRECTORY)
logger.info('Starting musdb evaluation (this could be long) ...')

View File

@@ -26,7 +26,7 @@ AudioInputArgument: ArgumentInfo = Argument(
resolve_path=True)
AudioAdapterOption: OptionInfo = Option(
'spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter',
'spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter',
'--adapter',
'-a',
help='Name of the audio adapter to use for audio I/O')

View File

@@ -72,10 +72,13 @@ def test_evaluate(backend):
with TemporaryDirectory() as directory:
generate_fake_eval_dataset(directory)
metrics = evaluate(
adapter='spleeter.audio.adapter.ffmpeg.FFMPEGProcessAudioAdapter',
output_path='eval',
stft_backend=backend,
params_filename='spleeter:4stems',
mus_dir=directory,
)
mwf=False,
verbose=False)
for instrument, metric in metrics.items():
for m, value in metric.items():
assert np.allclose(

View File

@@ -101,7 +101,6 @@ def test_train():
'-p', 'useless_config.json',
'-d', path
])
raise IOError(result.stdout)
assert result.exit_code == 0
# assert that model checkpoint was created.
assert os.path.exists(join(path, 'model', 'model.ckpt-10.index'))