mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
✅ fix evaluation testing
This commit is contained in:
@@ -69,19 +69,20 @@ def generate_fake_eval_dataset(path):
|
||||
|
||||
@pytest.mark.parametrize('backend', TEST_CONFIGURATIONS)
|
||||
def test_evaluate(backend):
|
||||
with TemporaryDirectory() as directory:
|
||||
generate_fake_eval_dataset(directory)
|
||||
metrics = evaluate(
|
||||
adapter='spleeter.audio.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(
|
||||
np.median(value),
|
||||
res_4stems[instrument][m],
|
||||
atol=1e-3)
|
||||
with TemporaryDirectory() as dataset:
|
||||
with TemporaryDirectory() as evaluation:
|
||||
generate_fake_eval_dataset(dataset)
|
||||
metrics = evaluate(
|
||||
adapter='spleeter.audio.ffmpeg.FFMPEGProcessAudioAdapter',
|
||||
output_path=evaluation,
|
||||
stft_backend=backend,
|
||||
params_filename='spleeter:4stems',
|
||||
mus_dir=dataset,
|
||||
mwf=False,
|
||||
verbose=False)
|
||||
for instrument, metric in metrics.items():
|
||||
for m, value in metric.items():
|
||||
assert np.allclose(
|
||||
np.median(value),
|
||||
res_4stems[instrument][m],
|
||||
atol=1e-3)
|
||||
|
||||
Reference in New Issue
Block a user