├── .gitignore ├── README.md ├── example ├── example.mp3 └── example_vocals.mp3 ├── images └── architecture.png ├── requirements.txt └── src ├── conf ├── augmentations │ ├── crop.yaml │ ├── crop_first.yaml │ ├── cropgainmix.yaml │ └── default.yaml ├── callbacks │ ├── default.yaml │ └── no_save.yaml ├── config.yaml ├── featurizer │ └── stft.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 ├── dataset.py ├── preprocessing.py └── utils.py ├── evaluate.py ├── files ├── vocals_train.txt └── vocals_valid.txt ├── inference.py ├── model ├── __init__.py ├── bandsplitrnn.py ├── modules │ ├── __init__.py │ ├── bandsequence.py │ ├── bandsplit.py │ ├── bandtransformer.py │ ├── maskestimation.py │ └── utils.py └── pl_model.py ├── prepare_dataset.py ├── saved_models ├── bass │ └── .gitkeep ├── drums │ └── .gitkeep ├── other │ └── .gitkeep └── vocals │ └── hparams.yaml ├── separator.py ├── train.py └── utils ├── callbacks.py ├── utils_inference.py └── utils_test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /example/example.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/example/example.mp3 -------------------------------------------------------------------------------- /example/example_vocals.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/example/example_vocals.mp3 -------------------------------------------------------------------------------- /images/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/images/architecture.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/conf/augmentations/crop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/augmentations/crop.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/crop_first.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/augmentations/crop_first.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/cropgainmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/augmentations/cropgainmix.yaml -------------------------------------------------------------------------------- /src/conf/augmentations/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/augmentations/default.yaml -------------------------------------------------------------------------------- /src/conf/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/callbacks/default.yaml -------------------------------------------------------------------------------- /src/conf/callbacks/no_save.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/callbacks/no_save.yaml -------------------------------------------------------------------------------- /src/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/config.yaml -------------------------------------------------------------------------------- /src/conf/featurizer/stft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/featurizer/stft.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnnV7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/model/bandsplitrnnV7.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnnbass.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/model/bandsplitrnnbass.yaml -------------------------------------------------------------------------------- /src/conf/model/bandsplitrnndrums.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/model/bandsplitrnndrums.yaml -------------------------------------------------------------------------------- /src/conf/sad/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/sad/default.yaml -------------------------------------------------------------------------------- /src/conf/test_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/test_dataset/default.yaml -------------------------------------------------------------------------------- /src/conf/train_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/train_dataset/default.yaml -------------------------------------------------------------------------------- /src/conf/train_dataset/no_augs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/train_dataset/no_augs.yaml -------------------------------------------------------------------------------- /src/conf/val_dataset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/conf/val_dataset/default.yaml -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/data/__init__.py -------------------------------------------------------------------------------- /src/data/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/data/augmentations.py -------------------------------------------------------------------------------- /src/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/data/dataset.py -------------------------------------------------------------------------------- /src/data/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/data/preprocessing.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/files/vocals_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/files/vocals_train.txt -------------------------------------------------------------------------------- /src/files/vocals_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/files/vocals_valid.txt -------------------------------------------------------------------------------- /src/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/inference.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/bandsplitrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/bandsplitrnn.py -------------------------------------------------------------------------------- /src/model/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/__init__.py -------------------------------------------------------------------------------- /src/model/modules/bandsequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/bandsequence.py -------------------------------------------------------------------------------- /src/model/modules/bandsplit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/bandsplit.py -------------------------------------------------------------------------------- /src/model/modules/bandtransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/bandtransformer.py -------------------------------------------------------------------------------- /src/model/modules/maskestimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/maskestimation.py -------------------------------------------------------------------------------- /src/model/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/modules/utils.py -------------------------------------------------------------------------------- /src/model/pl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/model/pl_model.py -------------------------------------------------------------------------------- /src/prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/prepare_dataset.py -------------------------------------------------------------------------------- /src/saved_models/bass/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/saved_models/drums/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/saved_models/other/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/saved_models/vocals/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/saved_models/vocals/hparams.yaml -------------------------------------------------------------------------------- /src/separator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/separator.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/utils/callbacks.py -------------------------------------------------------------------------------- /src/utils/utils_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/utils/utils_inference.py -------------------------------------------------------------------------------- /src/utils/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amanteur/BandSplitRNN-PyTorch/HEAD/src/utils/utils_test.py --------------------------------------------------------------------------------