├── .gitignore ├── .pre-commit-config.yaml ├── .project-root ├── LICENSE ├── Makefile ├── README.md ├── Resources ├── drb_SGMSE_GAN.png ├── drb_original.png ├── hybrid_2_SGMSE_GAN.png ├── hybrid_2_original.png ├── hybrid_3_SGMSE_GAN.png ├── hybrid_3_original.png ├── hybrid_4_SGMSE_GAN.png ├── hybrid_4_original.png ├── sqe_SGMSE_GAN.png ├── sqe_original.png └── video-demo.png ├── configs ├── __init__.py ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ ├── none.yaml │ └── rich_progress_bar.yaml ├── data │ ├── distort.yaml │ └── loadwav.yaml ├── debug │ ├── default.yaml │ ├── fdr.yaml │ ├── limit.yaml │ ├── overfit.yaml │ └── profiler.yaml ├── eval.yaml ├── experiment │ ├── LSGAN.yaml │ └── SGMSE_Large.yaml ├── extras │ └── default.yaml ├── hparams_search │ └── mnist_optuna.yaml ├── hydra │ └── default.yaml ├── local │ └── .gitkeep ├── logger │ ├── aim.yaml │ ├── comet.yaml │ ├── csv.yaml │ ├── many_loggers.yaml │ ├── mlflow.yaml │ ├── neptune.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ ├── LSGAN.yaml │ └── SGMSE_Large.yaml ├── paths │ └── default.yaml ├── predict.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ ├── mps.yaml │ └── universepp.yaml ├── pyproject.toml ├── requirements.txt ├── scripts ├── data_preparation │ ├── generate_ears_speech_list.py │ ├── list2json.py │ ├── predenoise_modelscope.py │ ├── predenoise_silero.py │ └── rir_gen │ │ ├── pyroomacoustics_scripts │ │ ├── rir_gen_monaural_48k_all_distances_farfield.py │ │ └── test_rir_gen.py │ │ ├── rir_gen_monaural_24k_all_distances_farfield.py │ │ ├── rir_gen_monaural_24k_all_distances_nearfield.py │ │ ├── rir_gen_monaural_48k_all_distances_farfield.py │ │ └── rir_gen_monaural_48k_all_distances_nearfield.py └── schedule.sh ├── src ├── __init__.py ├── data │ ├── __init__.py │ ├── components │ │ ├── FRA_RIR.py │ │ ├── __init__.py │ │ ├── collate.py │ │ ├── comm_distort_simu_dataset.py │ │ ├── loadwav_dataset.py │ │ ├── perturb.py │ │ └── webrtc_utils.py │ ├── distort_datamodule.py │ └── loadwav_datamodule.py ├── eval.py ├── models │ ├── LSGAN_module.py │ ├── SGMSE_module.py │ ├── __init__.py │ └── components │ │ ├── GAN │ │ ├── discriminator │ │ │ ├── __init__.py │ │ │ ├── hifigan │ │ │ │ ├── __init__.py │ │ │ │ ├── criteria.py │ │ │ │ ├── models.py │ │ │ │ └── open_models.py │ │ │ └── hifigan_vocoder │ │ │ │ ├── __init__.py │ │ │ │ ├── audio_torch.py │ │ │ │ ├── hifigan.py │ │ │ │ ├── hifigan_dicriminator.py │ │ │ │ ├── layers.py │ │ │ │ └── utils.py │ │ └── generator │ │ │ ├── CSMGAN │ │ │ ├── __init__.py │ │ │ └── generator5_24k.py │ │ │ ├── __init__.py │ │ │ └── ncsnpp │ │ │ ├── __init__.py │ │ │ └── model_wrapper.py │ │ ├── __init__.py │ │ ├── feature │ │ ├── __init__.py │ │ └── stft.py │ │ ├── loss_function │ │ └── monaural_loss.py │ │ └── sgmse │ │ ├── __init__.py │ │ ├── backbones │ │ ├── __init__.py │ │ ├── convtasnet.py │ │ ├── convtasnet_utils │ │ │ └── utils.py │ │ ├── gagnet.py │ │ ├── ncsnpp.py │ │ ├── ncsnpp_utils │ │ │ ├── layers.py │ │ │ ├── layerspp.py │ │ │ ├── normalization.py │ │ │ ├── op │ │ │ │ ├── __init__.py │ │ │ │ ├── fused_act.py │ │ │ │ ├── fused_bias_act.cpp │ │ │ │ ├── fused_bias_act_kernel.cu │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ ├── upfirdn2d.py │ │ │ │ └── upfirdn2d_kernel.cu │ │ │ ├── up_or_down_sampling.py │ │ │ └── utils.py │ │ └── shared.py │ │ ├── model.py │ │ ├── model_wrapper.py │ │ ├── sampling │ │ ├── __init__.py │ │ ├── correctors.py │ │ └── predictors.py │ │ ├── sdes.py │ │ └── util │ │ ├── graphics.py │ │ ├── inference.py │ │ ├── other.py │ │ ├── registry.py │ │ └── tensors.py ├── predict.py ├── train.py └── utils │ ├── __init__.py │ ├── instantiators.py │ ├── logging_utils.py │ ├── pylogger.py │ ├── rich_utils.py │ └── utils.py └── tests ├── __init__.py ├── helpers ├── __init__.py ├── package_available.py └── run_if.py └── test_perturb.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.project-root: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/.project-root -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/README.md -------------------------------------------------------------------------------- /Resources/drb_SGMSE_GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/drb_SGMSE_GAN.png -------------------------------------------------------------------------------- /Resources/drb_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/drb_original.png -------------------------------------------------------------------------------- /Resources/hybrid_2_SGMSE_GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_2_SGMSE_GAN.png -------------------------------------------------------------------------------- /Resources/hybrid_2_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_2_original.png -------------------------------------------------------------------------------- /Resources/hybrid_3_SGMSE_GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_3_SGMSE_GAN.png -------------------------------------------------------------------------------- /Resources/hybrid_3_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_3_original.png -------------------------------------------------------------------------------- /Resources/hybrid_4_SGMSE_GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_4_SGMSE_GAN.png -------------------------------------------------------------------------------- /Resources/hybrid_4_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/hybrid_4_original.png -------------------------------------------------------------------------------- /Resources/sqe_SGMSE_GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/sqe_SGMSE_GAN.png -------------------------------------------------------------------------------- /Resources/sqe_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/sqe_original.png -------------------------------------------------------------------------------- /Resources/video-demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/Resources/video-demo.png -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/data/distort.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/data/distort.yaml -------------------------------------------------------------------------------- /configs/data/loadwav.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/data/loadwav.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/fdr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/debug/fdr.yaml -------------------------------------------------------------------------------- /configs/debug/limit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/debug/limit.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/experiment/LSGAN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/experiment/LSGAN.yaml -------------------------------------------------------------------------------- /configs/experiment/SGMSE_Large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/experiment/SGMSE_Large.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hparams_search/mnist_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/hparams_search/mnist_optuna.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/local/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/logger/aim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/aim.yaml -------------------------------------------------------------------------------- /configs/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/comet.yaml -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/LSGAN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/model/LSGAN.yaml -------------------------------------------------------------------------------- /configs/model/SGMSE_Large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/model/SGMSE_Large.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/predict.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/predict.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /configs/trainer/universepp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/configs/trainer/universepp.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preparation/generate_ears_speech_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/generate_ears_speech_list.py -------------------------------------------------------------------------------- /scripts/data_preparation/list2json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/list2json.py -------------------------------------------------------------------------------- /scripts/data_preparation/predenoise_modelscope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/predenoise_modelscope.py -------------------------------------------------------------------------------- /scripts/data_preparation/predenoise_silero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/predenoise_silero.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/pyroomacoustics_scripts/rir_gen_monaural_48k_all_distances_farfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/pyroomacoustics_scripts/rir_gen_monaural_48k_all_distances_farfield.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/pyroomacoustics_scripts/test_rir_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/pyroomacoustics_scripts/test_rir_gen.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/rir_gen_monaural_24k_all_distances_farfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/rir_gen_monaural_24k_all_distances_farfield.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/rir_gen_monaural_24k_all_distances_nearfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/rir_gen_monaural_24k_all_distances_nearfield.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/rir_gen_monaural_48k_all_distances_farfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/rir_gen_monaural_48k_all_distances_farfield.py -------------------------------------------------------------------------------- /scripts/data_preparation/rir_gen/rir_gen_monaural_48k_all_distances_nearfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/data_preparation/rir_gen/rir_gen_monaural_48k_all_distances_nearfield.py -------------------------------------------------------------------------------- /scripts/schedule.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/scripts/schedule.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/FRA_RIR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/FRA_RIR.py -------------------------------------------------------------------------------- /src/data/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/collate.py -------------------------------------------------------------------------------- /src/data/components/comm_distort_simu_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/comm_distort_simu_dataset.py -------------------------------------------------------------------------------- /src/data/components/loadwav_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/loadwav_dataset.py -------------------------------------------------------------------------------- /src/data/components/perturb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/perturb.py -------------------------------------------------------------------------------- /src/data/components/webrtc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/components/webrtc_utils.py -------------------------------------------------------------------------------- /src/data/distort_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/distort_datamodule.py -------------------------------------------------------------------------------- /src/data/loadwav_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/data/loadwav_datamodule.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/LSGAN_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/LSGAN_module.py -------------------------------------------------------------------------------- /src/models/SGMSE_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/SGMSE_module.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan/criteria.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan/criteria.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan/models.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan/open_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan/open_models.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/audio_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan_vocoder/audio_torch.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/hifigan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan_vocoder/hifigan.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/hifigan_dicriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan_vocoder/hifigan_dicriminator.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan_vocoder/layers.py -------------------------------------------------------------------------------- /src/models/components/GAN/discriminator/hifigan_vocoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/discriminator/hifigan_vocoder/utils.py -------------------------------------------------------------------------------- /src/models/components/GAN/generator/CSMGAN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/generator/CSMGAN/generator5_24k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/generator/CSMGAN/generator5_24k.py -------------------------------------------------------------------------------- /src/models/components/GAN/generator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/generator/ncsnpp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/GAN/generator/ncsnpp/model_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/GAN/generator/ncsnpp/model_wrapper.py -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/feature/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/feature/stft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/feature/stft.py -------------------------------------------------------------------------------- /src/models/components/loss_function/monaural_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/loss_function/monaural_loss.py -------------------------------------------------------------------------------- /src/models/components/sgmse/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/__init__.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/convtasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/convtasnet.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/convtasnet_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/convtasnet_utils/utils.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/gagnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/gagnet.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/layers.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/layerspp.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/normalization.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/__init__.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_act.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/up_or_down_sampling.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/ncsnpp_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/ncsnpp_utils/utils.py -------------------------------------------------------------------------------- /src/models/components/sgmse/backbones/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/backbones/shared.py -------------------------------------------------------------------------------- /src/models/components/sgmse/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/model.py -------------------------------------------------------------------------------- /src/models/components/sgmse/model_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/model_wrapper.py -------------------------------------------------------------------------------- /src/models/components/sgmse/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/sampling/__init__.py -------------------------------------------------------------------------------- /src/models/components/sgmse/sampling/correctors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/sampling/correctors.py -------------------------------------------------------------------------------- /src/models/components/sgmse/sampling/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/sampling/predictors.py -------------------------------------------------------------------------------- /src/models/components/sgmse/sdes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/sdes.py -------------------------------------------------------------------------------- /src/models/components/sgmse/util/graphics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/util/graphics.py -------------------------------------------------------------------------------- /src/models/components/sgmse/util/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/util/inference.py -------------------------------------------------------------------------------- /src/models/components/sgmse/util/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/util/other.py -------------------------------------------------------------------------------- /src/models/components/sgmse/util/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/util/registry.py -------------------------------------------------------------------------------- /src/models/components/sgmse/util/tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/models/components/sgmse/util/tensors.py -------------------------------------------------------------------------------- /src/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/predict.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/instantiators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/instantiators.py -------------------------------------------------------------------------------- /src/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/logging_utils.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/package_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/tests/helpers/package_available.py -------------------------------------------------------------------------------- /tests/helpers/run_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/tests/helpers/run_if.py -------------------------------------------------------------------------------- /tests/test_perturb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanless/universal-speech-enhancement/HEAD/tests/test_perturb.py --------------------------------------------------------------------------------