├── .env.example ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── README_SUBMISSION.md ├── conda_env_gpu.yaml ├── configs ├── callbacks │ ├── default.yaml │ ├── none.yaml │ └── wandb.yaml ├── config.yaml ├── datamodule │ └── musdb18_hq.yaml ├── evaluation.yaml ├── experiment │ ├── debug.yaml │ ├── mixer.yaml │ ├── multigpu_bass.yaml │ ├── multigpu_default.yaml │ ├── multigpu_drums.yaml │ ├── multigpu_other.yaml │ └── multigpu_vocals.yaml ├── hydra │ └── default.yaml ├── logger │ ├── csv.yaml │ ├── many_loggers.yaml │ ├── neptune.yaml │ ├── none.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ ├── ConvTDFNet_bass.yaml │ ├── ConvTDFNet_drums.yaml │ ├── ConvTDFNet_other.yaml │ ├── ConvTDFNet_vocals.yaml │ └── Mixer.yaml └── trainer │ ├── default.yaml │ └── minimal.yaml ├── onnx_callback.png ├── requirements.txt ├── run.py ├── run_eval.py ├── setup.cfg ├── src ├── __init__.py ├── callbacks │ ├── __init__.py │ ├── onnx_callback.py │ └── wandb_callbacks.py ├── datamodules │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ └── musdb.py │ └── musdb_datamodule.py ├── evaluation │ ├── eval.py │ └── separate.py ├── mdx_kit │ ├── README.md │ └── evaluator │ │ ├── __init__.py │ │ ├── aicrowd_helpers.py │ │ └── music_demixing.py ├── models │ ├── __init__.py │ ├── mdxnet.py │ └── modules.py ├── train.py └── utils │ ├── __init__.py │ ├── data_augmentation.py │ └── utils.py ├── tests ├── __init__.py ├── helpers │ ├── __init__.py │ ├── module_available.py │ ├── run_command.py │ └── runif.py ├── smoke │ ├── __init__.py │ ├── test_commands.py │ ├── test_mixed_precision.py │ ├── test_sweeps.py │ └── test_wandb.py ├── submit │ └── to_onnx.py └── unit │ ├── __init__.py │ └── test_sth.py └── val_loss_vocals.png /.env.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/.env.example -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/README.md -------------------------------------------------------------------------------- /README_SUBMISSION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/README_SUBMISSION.md -------------------------------------------------------------------------------- /conda_env_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/conda_env_gpu.yaml -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/callbacks/wandb.yaml -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/datamodule/musdb18_hq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/datamodule/musdb18_hq.yaml -------------------------------------------------------------------------------- /configs/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/evaluation.yaml -------------------------------------------------------------------------------- /configs/experiment/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/debug.yaml -------------------------------------------------------------------------------- /configs/experiment/mixer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/mixer.yaml -------------------------------------------------------------------------------- /configs/experiment/multigpu_bass.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/multigpu_bass.yaml -------------------------------------------------------------------------------- /configs/experiment/multigpu_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/multigpu_default.yaml -------------------------------------------------------------------------------- /configs/experiment/multigpu_drums.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/multigpu_drums.yaml -------------------------------------------------------------------------------- /configs/experiment/multigpu_other.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/multigpu_other.yaml -------------------------------------------------------------------------------- /configs/experiment/multigpu_vocals.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/experiment/multigpu_vocals.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/logger/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/ConvTDFNet_bass.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/model/ConvTDFNet_bass.yaml -------------------------------------------------------------------------------- /configs/model/ConvTDFNet_drums.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/model/ConvTDFNet_drums.yaml -------------------------------------------------------------------------------- /configs/model/ConvTDFNet_other.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/model/ConvTDFNet_other.yaml -------------------------------------------------------------------------------- /configs/model/ConvTDFNet_vocals.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/model/ConvTDFNet_vocals.yaml -------------------------------------------------------------------------------- /configs/model/Mixer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/model/Mixer.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/minimal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/configs/trainer/minimal.yaml -------------------------------------------------------------------------------- /onnx_callback.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/onnx_callback.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/run.py -------------------------------------------------------------------------------- /run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/run_eval.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/setup.cfg -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/callbacks/onnx_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/callbacks/onnx_callback.py -------------------------------------------------------------------------------- /src/callbacks/wandb_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/callbacks/wandb_callbacks.py -------------------------------------------------------------------------------- /src/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/datasets/musdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/datamodules/datasets/musdb.py -------------------------------------------------------------------------------- /src/datamodules/musdb_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/datamodules/musdb_datamodule.py -------------------------------------------------------------------------------- /src/evaluation/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/evaluation/eval.py -------------------------------------------------------------------------------- /src/evaluation/separate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/evaluation/separate.py -------------------------------------------------------------------------------- /src/mdx_kit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/mdx_kit/README.md -------------------------------------------------------------------------------- /src/mdx_kit/evaluator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mdx_kit/evaluator/aicrowd_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/mdx_kit/evaluator/aicrowd_helpers.py -------------------------------------------------------------------------------- /src/mdx_kit/evaluator/music_demixing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/mdx_kit/evaluator/music_demixing.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/mdxnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/models/mdxnet.py -------------------------------------------------------------------------------- /src/models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/models/modules.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/utils/data_augmentation.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/module_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/helpers/module_available.py -------------------------------------------------------------------------------- /tests/helpers/run_command.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/helpers/run_command.py -------------------------------------------------------------------------------- /tests/helpers/runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/helpers/runif.py -------------------------------------------------------------------------------- /tests/smoke/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/smoke/test_commands.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/smoke/test_commands.py -------------------------------------------------------------------------------- /tests/smoke/test_mixed_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/smoke/test_mixed_precision.py -------------------------------------------------------------------------------- /tests/smoke/test_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/smoke/test_sweeps.py -------------------------------------------------------------------------------- /tests/smoke/test_wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/smoke/test_wandb.py -------------------------------------------------------------------------------- /tests/submit/to_onnx.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/test_sth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/tests/unit/test_sth.py -------------------------------------------------------------------------------- /val_loss_vocals.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/mdx-net/HEAD/val_loss_vocals.png --------------------------------------------------------------------------------