From d370a878fc234c4b064c62cb7182a077086dbbba Mon Sep 17 00:00:00 2001 From: Stefan Uhlich Date: Fri, 7 May 2021 09:52:54 +0200 Subject: [PATCH] Fix evaluation --- evaluator/music_demixing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/evaluator/music_demixing.py b/evaluator/music_demixing.py index a01ec03..574922b 100644 --- a/evaluator/music_demixing.py +++ b/evaluator/music_demixing.py @@ -91,7 +91,7 @@ def sdr(references, estimates): estimates = [] for instrument in instruments: reference_file = join(self.test_data_path, music_name, instrument + ".wav") - estimate_file = self.get_music_file_location(music_name, "bass") + estimate_file = self.get_music_file_location(music_name, instrument) reference, _ = sf.read(reference_file) estimate, _ = sf.read(estimate_file) references.append(reference)