yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing gltches issues with Istft

Browse Source
This commit is contained in:
mmoussallam
2020-06-18 18:01:03 +02:00
parent ca5cdd7d28
commit 27f52b71d5
3 changed files with 56 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/test_eval.py
View File

@@ -99,18 +99,14 @@ def generate_fake_eval_dataset(path):
aa.save(filename, data, fs)
@pytest.mark.parametrize('backend', TEST_CONFIGURATIONS)
def test_evaluate(backend):
with TemporaryDirectory() as directory:
generate_fake_eval_dataset(directory)
p = create_argument_parser()
arguments = p.parse_args(["evaluate", "-p", "spleeter:4stems", "--mus_dir", directory, "-B", backend])
params = load_configuration(arguments.configuration)
metrics = evaluate.entrypoint(arguments, params)
for instrument, metric in metrics.items():
for metric, value in metric.items():
assert np.allclose(np.median(value), res_4stems[backend][instrument][metric], atol=1e-3)
# test_evaluate("tensorflow")
def test_evaluate(path="FAKE_MUSDB_DIR"):
generate_fake_eval_dataset(path)
p = create_argument_parser()
arguments = p.parse_args(["evaluate", "-p", "spleeter:4stems", "--mus_dir", path])
params = load_configuration(arguments.configuration)
metrics = evaluate.entrypoint(arguments, params)
for instrument, metric in metrics.items():
print(instrument), print(metric)
for m, value in metric.items():
print(np.median(value)), print(res_4stems[instrument][m])
assert np.allclose(np.median(value), res_4stems[instrument][m], atol=1e-3)