├── LICENSE ├── README.md ├── calc_metrics.py ├── calc_metrics.sh ├── enhancement.py ├── enhancement.sh ├── nase.png ├── noise_labels ├── log_testset.txt └── log_trainset_28spk.txt ├── preprocessing ├── create_wsj0_chime3.py ├── create_wsj0_qut.py └── create_wsj0_reverb.py ├── requirements.txt ├── sgmse ├── BEATs.py ├── __pycache__ │ ├── BEATs.cpython-38.pyc │ ├── BEATs_ori.cpython-38.pyc │ ├── backbone.cpython-38.pyc │ ├── data_module.cpython-38.pyc │ ├── model.cpython-38.pyc │ ├── modules.cpython-38.pyc │ └── sdes.cpython-38.pyc ├── backbone.py ├── backbones │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── dcunet.cpython-38.pyc │ │ ├── ncsnpp.cpython-38.pyc │ │ └── shared.cpython-38.pyc │ ├── dcunet.py │ ├── ncsnpp.py │ ├── ncsnpp_utils │ │ ├── __pycache__ │ │ │ ├── layers.cpython-38.pyc │ │ │ ├── layerspp.cpython-38.pyc │ │ │ ├── normalization.cpython-38.pyc │ │ │ └── up_or_down_sampling.cpython-38.pyc │ │ ├── layers.py │ │ ├── layerspp.py │ │ ├── normalization.py │ │ ├── op │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-38.pyc │ │ │ │ ├── fused_act.cpython-38.pyc │ │ │ │ └── upfirdn2d.cpython-38.pyc │ │ │ ├── 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 ├── data_module.py ├── model.py ├── modules.py ├── sampling │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── correctors.cpython-38.pyc │ │ └── predictors.cpython-38.pyc │ ├── correctors.py │ └── predictors.py ├── sdes.py └── util │ ├── __pycache__ │ ├── inference.cpython-38.pyc │ ├── other.cpython-38.pyc │ ├── registry.cpython-38.pyc │ └── tensors.cpython-38.pyc │ ├── inference.py │ ├── other.py │ ├── registry.py │ └── tensors.py ├── train.py └── train.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/README.md -------------------------------------------------------------------------------- /calc_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/calc_metrics.py -------------------------------------------------------------------------------- /calc_metrics.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/calc_metrics.sh -------------------------------------------------------------------------------- /enhancement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/enhancement.py -------------------------------------------------------------------------------- /enhancement.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/enhancement.sh -------------------------------------------------------------------------------- /nase.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/nase.png -------------------------------------------------------------------------------- /noise_labels/log_testset.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/noise_labels/log_testset.txt -------------------------------------------------------------------------------- /noise_labels/log_trainset_28spk.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/noise_labels/log_trainset_28spk.txt -------------------------------------------------------------------------------- /preprocessing/create_wsj0_chime3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/preprocessing/create_wsj0_chime3.py -------------------------------------------------------------------------------- /preprocessing/create_wsj0_qut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/preprocessing/create_wsj0_qut.py -------------------------------------------------------------------------------- /preprocessing/create_wsj0_reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/preprocessing/create_wsj0_reverb.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/requirements.txt -------------------------------------------------------------------------------- /sgmse/BEATs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/BEATs.py -------------------------------------------------------------------------------- /sgmse/__pycache__/BEATs.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/BEATs.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/BEATs_ori.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/BEATs_ori.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/backbone.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/backbone.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/data_module.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/data_module.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/modules.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/modules.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/__pycache__/sdes.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/__pycache__/sdes.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbone.py -------------------------------------------------------------------------------- /sgmse/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/__init__.py -------------------------------------------------------------------------------- /sgmse/backbones/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/__pycache__/dcunet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/__pycache__/dcunet.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/__pycache__/ncsnpp.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/__pycache__/ncsnpp.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/__pycache__/shared.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/__pycache__/shared.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/dcunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/dcunet.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/__pycache__/layers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/__pycache__/layers.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/__pycache__/layerspp.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/__pycache__/layerspp.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/__pycache__/normalization.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/__pycache__/normalization.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/__pycache__/up_or_down_sampling.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/__pycache__/up_or_down_sampling.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/layers.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/layerspp.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/normalization.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/__init__.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/__pycache__/fused_act.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/__pycache__/fused_act.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/fused_act.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/upfirdn2d.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/up_or_down_sampling.py -------------------------------------------------------------------------------- /sgmse/backbones/ncsnpp_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/ncsnpp_utils/utils.py -------------------------------------------------------------------------------- /sgmse/backbones/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/backbones/shared.py -------------------------------------------------------------------------------- /sgmse/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/data_module.py -------------------------------------------------------------------------------- /sgmse/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/model.py -------------------------------------------------------------------------------- /sgmse/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/modules.py -------------------------------------------------------------------------------- /sgmse/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/__init__.py -------------------------------------------------------------------------------- /sgmse/sampling/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/sampling/__pycache__/correctors.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/__pycache__/correctors.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/sampling/__pycache__/predictors.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/__pycache__/predictors.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/sampling/correctors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/correctors.py -------------------------------------------------------------------------------- /sgmse/sampling/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sampling/predictors.py -------------------------------------------------------------------------------- /sgmse/sdes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/sdes.py -------------------------------------------------------------------------------- /sgmse/util/__pycache__/inference.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/__pycache__/inference.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/util/__pycache__/other.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/__pycache__/other.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/util/__pycache__/registry.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/__pycache__/registry.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/util/__pycache__/tensors.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/__pycache__/tensors.cpython-38.pyc -------------------------------------------------------------------------------- /sgmse/util/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/inference.py -------------------------------------------------------------------------------- /sgmse/util/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/other.py -------------------------------------------------------------------------------- /sgmse/util/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/registry.py -------------------------------------------------------------------------------- /sgmse/util/tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/sgmse/util/tensors.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YUCHEN005/NASE/HEAD/train.sh --------------------------------------------------------------------------------