├── .gitignore ├── README.md ├── eval.sh ├── eval_general.py ├── local ├── ConvTasNet.yml ├── DPRNNTasNet.yml ├── DPTNet.yml ├── SepFormer2TasNet.yml ├── SepFormerTasNet.yml ├── librimix │ ├── create_local_metadata.py │ └── prepare_data.sh └── wham │ ├── convert_sphere2wav.sh │ ├── prepare_data.sh │ └── preprocess_wham.py ├── perm_general.py ├── prepare_librimix_data.sh ├── prepare_wham_data.sh ├── requirements.txt ├── run.sh ├── scripts ├── run_ConvTasNet_Libri2Mix_sep_clean_from_scratch.sh ├── run_ConvTasNet_Libri2Mix_sep_clean_multi_task.sh ├── run_ConvTasNet_Libri2Mix_sep_clean_pretrained.sh ├── run_ConvTasNet_wsj0-2mix_sep_clean_from_scratch.sh ├── run_ConvTasNet_wsj0-2mix_sep_clean_multi_task.sh ├── run_ConvTasNet_wsj0-2mix_sep_clean_pretrained.sh ├── run_DPRNN_Libri2Mix_enh_single_from_scratch.sh ├── run_DPRNN_wsj0_sep_clean_from_scratch.sh ├── run_DPRNN_wsj0_sep_clean_pretrained.sh ├── run_DPTNet_Libri2Mix_enh_single_from_scratch.sh ├── run_DPTNet_wsj0_sep_clean_from_scratch.sh └── run_DPTNet_wsj0_sep_clean_pretrained.sh ├── src ├── __init__.py ├── data │ ├── __init__.py │ └── utils.py ├── engine │ └── system.py ├── losses │ └── multi_task_wrapper.py ├── masknn │ ├── __init__.py │ └── attention.py └── models │ ├── __init__.py │ └── sepformer_tasnet.py ├── train_general.py └── utils ├── parse_options.sh └── prepare_python_env.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/README.md -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/eval.sh -------------------------------------------------------------------------------- /eval_general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/eval_general.py -------------------------------------------------------------------------------- /local/ConvTasNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/ConvTasNet.yml -------------------------------------------------------------------------------- /local/DPRNNTasNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/DPRNNTasNet.yml -------------------------------------------------------------------------------- /local/DPTNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/DPTNet.yml -------------------------------------------------------------------------------- /local/SepFormer2TasNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/SepFormer2TasNet.yml -------------------------------------------------------------------------------- /local/SepFormerTasNet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/SepFormerTasNet.yml -------------------------------------------------------------------------------- /local/librimix/create_local_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/librimix/create_local_metadata.py -------------------------------------------------------------------------------- /local/librimix/prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/librimix/prepare_data.sh -------------------------------------------------------------------------------- /local/wham/convert_sphere2wav.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/wham/convert_sphere2wav.sh -------------------------------------------------------------------------------- /local/wham/prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/wham/prepare_data.sh -------------------------------------------------------------------------------- /local/wham/preprocess_wham.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/local/wham/preprocess_wham.py -------------------------------------------------------------------------------- /perm_general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/perm_general.py -------------------------------------------------------------------------------- /prepare_librimix_data.sh: -------------------------------------------------------------------------------- 1 | local/librimix/prepare_data.sh -------------------------------------------------------------------------------- /prepare_wham_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/prepare_wham_data.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/run.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_Libri2Mix_sep_clean_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_Libri2Mix_sep_clean_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_Libri2Mix_sep_clean_multi_task.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_Libri2Mix_sep_clean_multi_task.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_Libri2Mix_sep_clean_pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_Libri2Mix_sep_clean_pretrained.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_wsj0-2mix_sep_clean_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_wsj0-2mix_sep_clean_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_wsj0-2mix_sep_clean_multi_task.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_wsj0-2mix_sep_clean_multi_task.sh -------------------------------------------------------------------------------- /scripts/run_ConvTasNet_wsj0-2mix_sep_clean_pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_ConvTasNet_wsj0-2mix_sep_clean_pretrained.sh -------------------------------------------------------------------------------- /scripts/run_DPRNN_Libri2Mix_enh_single_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPRNN_Libri2Mix_enh_single_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_DPRNN_wsj0_sep_clean_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPRNN_wsj0_sep_clean_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_DPRNN_wsj0_sep_clean_pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPRNN_wsj0_sep_clean_pretrained.sh -------------------------------------------------------------------------------- /scripts/run_DPTNet_Libri2Mix_enh_single_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPTNet_Libri2Mix_enh_single_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_DPTNet_wsj0_sep_clean_from_scratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPTNet_wsj0_sep_clean_from_scratch.sh -------------------------------------------------------------------------------- /scripts/run_DPTNet_wsj0_sep_clean_pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/scripts/run_DPTNet_wsj0_sep_clean_pretrained.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/data/__init__.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/engine/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/engine/system.py -------------------------------------------------------------------------------- /src/losses/multi_task_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/losses/multi_task_wrapper.py -------------------------------------------------------------------------------- /src/masknn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/masknn/__init__.py -------------------------------------------------------------------------------- /src/masknn/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/masknn/attention.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/sepformer_tasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/src/models/sepformer_tasnet.py -------------------------------------------------------------------------------- /train_general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/train_general.py -------------------------------------------------------------------------------- /utils/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/utils/parse_options.sh -------------------------------------------------------------------------------- /utils/prepare_python_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SungFeng-Huang/SSL-pretraining-separation/HEAD/utils/prepare_python_env.sh --------------------------------------------------------------------------------