Skip to content

Commit 0978064

Browse files
authoredDec 17, 2024
Begin adding evaluation scores for all models, initially MUSEDB18 + model fix (#169)
* Added initial attempt at script to evaluate model performance using MUSDB18 dataset and evaluation tooling * Added loading of existing raw scores * WIP combined scores * Updated script to process all tracks for all models, and evaluate SDR for all stems supported by MUSDB18. Added combined scores calculated so far. * Added median score calculation, changed inner/outer loop for tracks vs. model processing, renamed accompaniment * Fixed UVR-De-Reverb-aufr33-jarredou.pth model by adding new model-data dict appended to the UVR one
1 parent 88769c3 commit 0978064

File tree

7 files changed

+5550
-84
lines changed

7 files changed

+5550
-84
lines changed
 

‎.gitignore

+2
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
*.wav
1212
*.flac
1313
*.mp3
14+
tests/model-metrics/results
15+
tests/model-metrics/datasets
1416

1517
# Byte-compiled / optimized / DLL files
1618
__pycache__/

‎audio_separator/model-data.json

+14
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
{
2+
"vr_model_data": {
3+
"97dc361a7a88b2c4542f68364b32c7f6": {
4+
"vr_model_param": "4band_v4_ms_fullband",
5+
"primary_stem": "Dry",
6+
"nout": 32,
7+
"nout_lstm": 128,
8+
"is_karaoke": false,
9+
"is_bv_model": false,
10+
"is_bv_model_rebalanced": 0.0
11+
}
12+
},
13+
"mdx_model_data": {}
14+
}

0 commit comments

Comments
 (0)