├── .gitattributes ├── .gitignore ├── README.md ├── images ├── CL Banner.png ├── Source Separation Methods Over Time.png ├── architecture_full.png ├── architecture_paper.png ├── band-split_module.png ├── band_and_sequence_separation_modeling_module.png ├── competition_results.png ├── mask_estimation_module.png ├── source_separation_methods.png ├── spectrograms.png ├── spectrums.png └── waveforms.png ├── presentation └── Music Demixing Presentation.pdf ├── requirements.txt └── src ├── conf ├── augmentations │ ├── crop.yaml │ ├── crop_first.yaml │ ├── cropgainmix.yaml │ └── default.yaml ├── callbacks │ ├── default.yaml │ ├── no_save.yaml │ └── wandb.yaml ├── config.yaml ├── featurizer │ └── stft.yaml ├── logger │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ ├── bandsplitrnnV7.yaml │ ├── bandsplitrnnbass.yaml │ └── bandsplitrnndrums.yaml ├── sad │ └── default.yaml ├── test_dataset │ └── default.yaml ├── train_dataset │ ├── default.yaml │ └── no_augs.yaml └── val_dataset │ └── default.yaml ├── data ├── __init__.py ├── augmentations.py ├── combine_audio.py ├── copy_files.py ├── dataset.py ├── low_pass.py ├── preprocessing.py └── utils.py ├── evaluate.py ├── files ├── A_Label_Noise │ ├── bass_train.txt │ ├── drums_train.txt │ ├── other_train.txt │ └── vocals_train.txt ├── Full (no test) │ ├── vocals_train.txt │ └── vocals_valid.txt ├── Full │ ├── vocals_test.txt │ ├── vocals_train.txt │ └── vocals_valid.txt ├── Sample │ ├── vocals_test.txt │ ├── vocals_train.txt │ └── vocals_valid.txt ├── bass_train.txt ├── drums_train.txt ├── other_train.txt └── vocals_train.txt ├── inference.py ├── model ├── __init__.py ├── bandsplitrnn.py ├── modules │ ├── __init__.py │ ├── bandsequence.py │ ├── bandsplit.py │ ├── bandtransformer.py │ ├── maskestimation.py │ └── utils.py └── pl_model.py ├── notebooks ├── 01_Audio Stems Mixing.ipynb └── 02_EDA.ipynb ├── prepare_dataset.py ├── saved_models ├── bass │ └── hparams.yaml ├── drums │ └── hparams.yaml ├── other │ └── hparams.yaml └── vocals │ └── hparams.yaml ├── separator.py ├── train.py └── utils ├── __init__.py ├── callbacks.py ├── pylogger.py ├── utils_inference.py ├── utils_test.py └── utils_wandb.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/README.md -------------------------------------------------------------------------------- /images/CL Banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/CL Banner.png -------------------------------------------------------------------------------- /images/Source Separation Methods Over Time.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/Source Separation Methods Over Time.png -------------------------------------------------------------------------------- /images/architecture_full.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/architecture_full.png -------------------------------------------------------------------------------- /images/architecture_paper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/architecture_paper.png -------------------------------------------------------------------------------- /images/band-split_module.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/band-split_module.png -------------------------------------------------------------------------------- /images/band_and_sequence_separation_modeling_module.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/band_and_sequence_separation_modeling_module.png -------------------------------------------------------------------------------- /images/competition_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/competition_results.png -------------------------------------------------------------------------------- /images/mask_estimation_module.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/mask_estimation_module.png -------------------------------------------------------------------------------- /images/source_separation_methods.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/source_separation_methods.png -------------------------------------------------------------------------------- /images/spectrograms.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/spectrograms.png -------------------------------------------------------------------------------- /images/spectrums.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/spectrums.png -------------------------------------------------------------------------------- /images/waveforms.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/images/waveforms.png -------------------------------------------------------------------------------- /presentation/Music Demixing Presentation.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/presentation/Music Demixing Presentation.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/conf/augmentations/crop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/augmentations/crop.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/crop_first.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/augmentations/crop_first.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/cropgainmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/augmentations/cropgainmix.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/augmentations/default.yaml -------------------------------------------------------------------------------- /src/conf/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/callbacks/default.yaml -------------------------------------------------------------------------------- /src/conf/callbacks/no_save.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/callbacks/no_save.yaml -------------------------------------------------------------------------------- /src/conf/callbacks/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/callbacks/wandb.yaml -------------------------------------------------------------------------------- /src/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/config.yaml -------------------------------------------------------------------------------- /src/conf/featurizer/stft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/featurizer/stft.yaml -------------------------------------------------------------------------------- /src/conf/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/logger/tensorboard.yaml -------------------------------------------------------------------------------- /src/conf/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/logger/wandb.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnnV7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/model/bandsplitrnnV7.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnnbass.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/model/bandsplitrnnbass.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnndrums.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/model/bandsplitrnndrums.yaml -------------------------------------------------------------------------------- /src/conf/sad/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/sad/default.yaml -------------------------------------------------------------------------------- /src/conf/test_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/test_dataset/default.yaml -------------------------------------------------------------------------------- /src/conf/train_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/train_dataset/default.yaml -------------------------------------------------------------------------------- /src/conf/train_dataset/no_augs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/train_dataset/no_augs.yaml -------------------------------------------------------------------------------- /src/conf/val_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/conf/val_dataset/default.yaml -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/__init__.py -------------------------------------------------------------------------------- /src/data/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/augmentations.py -------------------------------------------------------------------------------- /src/data/combine_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/combine_audio.py -------------------------------------------------------------------------------- /src/data/copy_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/copy_files.py -------------------------------------------------------------------------------- /src/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/dataset.py -------------------------------------------------------------------------------- /src/data/low_pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/low_pass.py -------------------------------------------------------------------------------- /src/data/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/preprocessing.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/files/A_Label_Noise/bass_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/A_Label_Noise/bass_train.txt -------------------------------------------------------------------------------- /src/files/A_Label_Noise/drums_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/A_Label_Noise/drums_train.txt -------------------------------------------------------------------------------- /src/files/A_Label_Noise/other_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/A_Label_Noise/other_train.txt -------------------------------------------------------------------------------- /src/files/A_Label_Noise/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/A_Label_Noise/vocals_train.txt -------------------------------------------------------------------------------- /src/files/Full (no test)/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Full (no test)/vocals_train.txt -------------------------------------------------------------------------------- /src/files/Full (no test)/vocals_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Full (no test)/vocals_valid.txt -------------------------------------------------------------------------------- /src/files/Full/vocals_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Full/vocals_test.txt -------------------------------------------------------------------------------- /src/files/Full/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Full/vocals_train.txt -------------------------------------------------------------------------------- /src/files/Full/vocals_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Full/vocals_valid.txt -------------------------------------------------------------------------------- /src/files/Sample/vocals_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Sample/vocals_test.txt -------------------------------------------------------------------------------- /src/files/Sample/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Sample/vocals_train.txt -------------------------------------------------------------------------------- /src/files/Sample/vocals_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/Sample/vocals_valid.txt -------------------------------------------------------------------------------- /src/files/bass_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/bass_train.txt -------------------------------------------------------------------------------- /src/files/drums_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/drums_train.txt -------------------------------------------------------------------------------- /src/files/other_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/other_train.txt -------------------------------------------------------------------------------- /src/files/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/files/vocals_train.txt -------------------------------------------------------------------------------- /src/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/inference.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/bandsplitrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/bandsplitrnn.py -------------------------------------------------------------------------------- /src/model/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/__init__.py -------------------------------------------------------------------------------- /src/model/modules/bandsequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/bandsequence.py -------------------------------------------------------------------------------- /src/model/modules/bandsplit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/bandsplit.py -------------------------------------------------------------------------------- /src/model/modules/bandtransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/bandtransformer.py -------------------------------------------------------------------------------- /src/model/modules/maskestimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/maskestimation.py -------------------------------------------------------------------------------- /src/model/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/modules/utils.py -------------------------------------------------------------------------------- /src/model/pl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/model/pl_model.py -------------------------------------------------------------------------------- /src/notebooks/01_Audio Stems Mixing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/notebooks/01_Audio Stems Mixing.ipynb -------------------------------------------------------------------------------- /src/notebooks/02_EDA.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/notebooks/02_EDA.ipynb -------------------------------------------------------------------------------- /src/prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/prepare_dataset.py -------------------------------------------------------------------------------- /src/saved_models/bass/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/saved_models/bass/hparams.yaml -------------------------------------------------------------------------------- /src/saved_models/drums/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/saved_models/drums/hparams.yaml -------------------------------------------------------------------------------- /src/saved_models/other/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/saved_models/other/hparams.yaml -------------------------------------------------------------------------------- /src/saved_models/vocals/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/saved_models/vocals/hparams.yaml -------------------------------------------------------------------------------- /src/separator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/separator.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/callbacks.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/utils_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/utils_inference.py -------------------------------------------------------------------------------- /src/utils/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/utils_test.py -------------------------------------------------------------------------------- /src/utils/utils_wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crlandsc/Music-Demixing-with-Band-Split-RNN/HEAD/src/utils/utils_wandb.py --------------------------------------------------------------------------------