├── .gitignore ├── LICENSE ├── README.md ├── ckpt ├── .gitkeep └── README.md ├── data ├── README.md ├── export_libri2mix_funcodec.sh ├── export_librispeech_funcodec_normalize.sh ├── generate_list.py ├── libri2mix_100_360_aux_s1.txt └── utils │ ├── export_libri2mix_funcodec.py │ └── parse_options.sh ├── evaluation ├── README.md ├── evaluation.sh └── libri2mix_whisper_base.txt ├── exp ├── README.md ├── libri2mix │ └── config_log_mel_aux_5s.yaml ├── libri2mix_finetune │ └── config_log_mel_aux_5s_finetune_e_20.yaml └── librispeech │ └── config_log_mel_aux_5s_e_100_patience.yaml ├── log ├── .gitkeep └── README.md ├── recipes ├── inference.sh ├── log │ └── .gitkeep ├── run_tse_libri2mix.sh ├── run_tse_libri2mix_finetune.sh └── run_tse_librispeech_dm.sh ├── requirements.txt ├── src ├── _funcodec.py ├── _funcodec_data │ ├── _espnet_dataset.py │ ├── build_sequence_iter.py │ └── data_loaders.py ├── bin │ └── tse_inference.py ├── eval │ ├── dnsmos.py │ ├── speech_bert.py │ ├── wavlm_base_plus_sv_spksim_eval.py │ ├── wer.py │ ├── wer_direct.py │ └── wespeaker_eval.py ├── infer.py ├── model │ └── laura_model_only_clean.py ├── schedulers │ └── patience.py ├── train.py ├── trainer │ ├── helper.py │ └── trainer.py └── utils │ ├── audio.py │ ├── dprint.py │ ├── hinter.py │ ├── load_scp.py │ ├── mel_spectrogram.py │ ├── postprocess.py │ ├── rir_utils.py │ └── utils.py └── utils ├── nisqa_merge.py └── parse_options.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/.gitkeep: -------------------------------------------------------------------------------- 1 | CKPT will be output here. -------------------------------------------------------------------------------- /ckpt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/ckpt/README.md -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/README.md -------------------------------------------------------------------------------- /data/export_libri2mix_funcodec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/export_libri2mix_funcodec.sh -------------------------------------------------------------------------------- /data/export_librispeech_funcodec_normalize.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/export_librispeech_funcodec_normalize.sh -------------------------------------------------------------------------------- /data/generate_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/generate_list.py -------------------------------------------------------------------------------- /data/libri2mix_100_360_aux_s1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/libri2mix_100_360_aux_s1.txt -------------------------------------------------------------------------------- /data/utils/export_libri2mix_funcodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/utils/export_libri2mix_funcodec.py -------------------------------------------------------------------------------- /data/utils/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/data/utils/parse_options.sh -------------------------------------------------------------------------------- /evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/evaluation/README.md -------------------------------------------------------------------------------- /evaluation/evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/evaluation/evaluation.sh -------------------------------------------------------------------------------- /evaluation/libri2mix_whisper_base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/evaluation/libri2mix_whisper_base.txt -------------------------------------------------------------------------------- /exp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/exp/README.md -------------------------------------------------------------------------------- /exp/libri2mix/config_log_mel_aux_5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/exp/libri2mix/config_log_mel_aux_5s.yaml -------------------------------------------------------------------------------- /exp/libri2mix_finetune/config_log_mel_aux_5s_finetune_e_20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/exp/libri2mix_finetune/config_log_mel_aux_5s_finetune_e_20.yaml -------------------------------------------------------------------------------- /exp/librispeech/config_log_mel_aux_5s_e_100_patience.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/exp/librispeech/config_log_mel_aux_5s_e_100_patience.yaml -------------------------------------------------------------------------------- /log/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /log/README.md: -------------------------------------------------------------------------------- 1 | LOG FILE FOR TRAINING WILL BE OUTPUT HERE. -------------------------------------------------------------------------------- /recipes/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/recipes/inference.sh -------------------------------------------------------------------------------- /recipes/log/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recipes/run_tse_libri2mix.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/recipes/run_tse_libri2mix.sh -------------------------------------------------------------------------------- /recipes/run_tse_libri2mix_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/recipes/run_tse_libri2mix_finetune.sh -------------------------------------------------------------------------------- /recipes/run_tse_librispeech_dm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/recipes/run_tse_librispeech_dm.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/_funcodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/_funcodec.py -------------------------------------------------------------------------------- /src/_funcodec_data/_espnet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/_funcodec_data/_espnet_dataset.py -------------------------------------------------------------------------------- /src/_funcodec_data/build_sequence_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/_funcodec_data/build_sequence_iter.py -------------------------------------------------------------------------------- /src/_funcodec_data/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/_funcodec_data/data_loaders.py -------------------------------------------------------------------------------- /src/bin/tse_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/bin/tse_inference.py -------------------------------------------------------------------------------- /src/eval/dnsmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/dnsmos.py -------------------------------------------------------------------------------- /src/eval/speech_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/speech_bert.py -------------------------------------------------------------------------------- /src/eval/wavlm_base_plus_sv_spksim_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/wavlm_base_plus_sv_spksim_eval.py -------------------------------------------------------------------------------- /src/eval/wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/wer.py -------------------------------------------------------------------------------- /src/eval/wer_direct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/wer_direct.py -------------------------------------------------------------------------------- /src/eval/wespeaker_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/eval/wespeaker_eval.py -------------------------------------------------------------------------------- /src/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/infer.py -------------------------------------------------------------------------------- /src/model/laura_model_only_clean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/model/laura_model_only_clean.py -------------------------------------------------------------------------------- /src/schedulers/patience.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/schedulers/patience.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/train.py -------------------------------------------------------------------------------- /src/trainer/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/trainer/helper.py -------------------------------------------------------------------------------- /src/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/trainer/trainer.py -------------------------------------------------------------------------------- /src/utils/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/audio.py -------------------------------------------------------------------------------- /src/utils/dprint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/dprint.py -------------------------------------------------------------------------------- /src/utils/hinter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/hinter.py -------------------------------------------------------------------------------- /src/utils/load_scp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/load_scp.py -------------------------------------------------------------------------------- /src/utils/mel_spectrogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/mel_spectrogram.py -------------------------------------------------------------------------------- /src/utils/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/postprocess.py -------------------------------------------------------------------------------- /src/utils/rir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/rir_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /utils/nisqa_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/utils/nisqa_merge.py -------------------------------------------------------------------------------- /utils/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Beilong-Tang/lauraTSE_code/HEAD/utils/parse_options.sh --------------------------------------------------------------------------------