Skip to content

Commit

Permalink
Begin adding evaluation scores for all models, initially MUSEDB18 + m…
Browse files Browse the repository at this point in the history
…odel 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
  • Loading branch information
beveradb authored Dec 17, 2024
1 parent 88769c3 commit 0978064
Show file tree
Hide file tree
Showing 7 changed files with 5,550 additions and 84 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@
*.wav
*.flac
*.mp3
tests/model-metrics/results
tests/model-metrics/datasets

# Byte-compiled / optimized / DLL files
__pycache__/
Expand Down
14 changes: 14 additions & 0 deletions audio_separator/model-data.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
{
"vr_model_data": {
"97dc361a7a88b2c4542f68364b32c7f6": {
"vr_model_param": "4band_v4_ms_fullband",
"primary_stem": "Dry",
"nout": 32,
"nout_lstm": 128,
"is_karaoke": false,
"is_bv_model": false,
"is_bv_model_rebalanced": 0.0
}
},
"mdx_model_data": {}
}
Loading

0 comments on commit 0978064

Please sign in to comment.