├── .gitignore ├── README.md ├── audio_examples ├── clean │ ├── p226 │ │ └── p226_003.wav │ └── p287 │ │ └── p287_005.wav ├── reverberant │ ├── p226 │ │ └── p226_003.wav │ └── p287 │ │ └── p287_005.wav └── rir │ ├── p226 │ └── p226_003.wav │ └── p287 │ └── p287_005.wav ├── conf ├── conf_VCTK.yaml ├── diff_params │ ├── diff_params_op │ └── edm_VCTK.yaml ├── dset │ ├── vctk_16k_4s.yaml │ └── vctk_16k_4s_test-benchmark.yaml ├── exp │ └── VCTK_16k_4s_time.yaml ├── logging │ └── base_logging.yaml ├── network │ └── ncsnpp.yaml └── tester │ ├── blind_dereverberation_BUDDy.yaml │ ├── informed_dereverberation_DPS.yaml │ └── only_unconditional.yaml ├── datasets └── vctk.py ├── diff_params ├── __init__.py ├── edm.py └── shared.py ├── experiments └── .gitkeep ├── networks ├── ncsnpp.py └── ncsnpp_utils │ ├── layers.py │ ├── layerspp.py │ ├── normalization.py │ ├── op │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-311.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── fused_act.cpython-38.pyc │ │ ├── upfirdn2d.cpython-311.pyc │ │ ├── upfirdn2d.cpython-38.pyc │ │ └── upfirdn2d.cpython-39.pyc │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu │ ├── up_or_down_sampling.py │ └── utils.py ├── requirements.txt ├── test.py ├── test_blind_dereverberation.sh ├── test_informed_dereverberation.sh ├── testing ├── EulerHeunSampler.py ├── EulerHeunSamplerDPS.py ├── Sampler.py ├── operators │ ├── reverb.py │ ├── shared.py │ └── subband_filtering.py └── tester.py ├── train.py ├── train.sh ├── training └── trainer.py └── utils ├── log.py ├── losses.py ├── reverb_utils.py ├── setup.py ├── tensor_utils.py ├── torch_utils ├── __init__.py ├── misc.py └── training_stats.py └── training_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/README.md -------------------------------------------------------------------------------- /audio_examples/clean/p226/p226_003.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/clean/p226/p226_003.wav -------------------------------------------------------------------------------- /audio_examples/clean/p287/p287_005.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/clean/p287/p287_005.wav -------------------------------------------------------------------------------- /audio_examples/reverberant/p226/p226_003.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/reverberant/p226/p226_003.wav -------------------------------------------------------------------------------- /audio_examples/reverberant/p287/p287_005.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/reverberant/p287/p287_005.wav -------------------------------------------------------------------------------- /audio_examples/rir/p226/p226_003.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/rir/p226/p226_003.wav -------------------------------------------------------------------------------- /audio_examples/rir/p287/p287_005.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/audio_examples/rir/p287/p287_005.wav -------------------------------------------------------------------------------- /conf/conf_VCTK.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/conf_VCTK.yaml -------------------------------------------------------------------------------- /conf/diff_params/diff_params_op: -------------------------------------------------------------------------------- 1 | diff_params_op -------------------------------------------------------------------------------- /conf/diff_params/edm_VCTK.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/diff_params/edm_VCTK.yaml -------------------------------------------------------------------------------- /conf/dset/vctk_16k_4s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/dset/vctk_16k_4s.yaml -------------------------------------------------------------------------------- /conf/dset/vctk_16k_4s_test-benchmark.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/dset/vctk_16k_4s_test-benchmark.yaml -------------------------------------------------------------------------------- /conf/exp/VCTK_16k_4s_time.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/exp/VCTK_16k_4s_time.yaml -------------------------------------------------------------------------------- /conf/logging/base_logging.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/logging/base_logging.yaml -------------------------------------------------------------------------------- /conf/network/ncsnpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/network/ncsnpp.yaml -------------------------------------------------------------------------------- /conf/tester/blind_dereverberation_BUDDy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/tester/blind_dereverberation_BUDDy.yaml -------------------------------------------------------------------------------- /conf/tester/informed_dereverberation_DPS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/tester/informed_dereverberation_DPS.yaml -------------------------------------------------------------------------------- /conf/tester/only_unconditional.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/conf/tester/only_unconditional.yaml -------------------------------------------------------------------------------- /datasets/vctk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/datasets/vctk.py -------------------------------------------------------------------------------- /diff_params/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/diff_params/__init__.py -------------------------------------------------------------------------------- /diff_params/edm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/diff_params/edm.py -------------------------------------------------------------------------------- /diff_params/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/diff_params/shared.py -------------------------------------------------------------------------------- /experiments/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/layers.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/layerspp.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/normalization.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__init__.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/__init__.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/__init__.cpython-311.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/fused_act.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/fused_act.cpython-38.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-311.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-38.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/__pycache__/upfirdn2d.cpython-39.pyc -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/upfirdn2d.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /networks/ncsnpp_utils/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/up_or_down_sampling.py -------------------------------------------------------------------------------- /networks/ncsnpp_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/networks/ncsnpp_utils/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/test.py -------------------------------------------------------------------------------- /test_blind_dereverberation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/test_blind_dereverberation.sh -------------------------------------------------------------------------------- /test_informed_dereverberation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/test_informed_dereverberation.sh -------------------------------------------------------------------------------- /testing/EulerHeunSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/EulerHeunSampler.py -------------------------------------------------------------------------------- /testing/EulerHeunSamplerDPS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/EulerHeunSamplerDPS.py -------------------------------------------------------------------------------- /testing/Sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/Sampler.py -------------------------------------------------------------------------------- /testing/operators/reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/operators/reverb.py -------------------------------------------------------------------------------- /testing/operators/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/operators/shared.py -------------------------------------------------------------------------------- /testing/operators/subband_filtering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/operators/subband_filtering.py -------------------------------------------------------------------------------- /testing/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/testing/tester.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/train.sh -------------------------------------------------------------------------------- /training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/training/trainer.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/reverb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/reverb_utils.py -------------------------------------------------------------------------------- /utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/setup.py -------------------------------------------------------------------------------- /utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/tensor_utils.py -------------------------------------------------------------------------------- /utils/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/torch_utils/__init__.py -------------------------------------------------------------------------------- /utils/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/torch_utils/misc.py -------------------------------------------------------------------------------- /utils/torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/torch_utils/training_stats.py -------------------------------------------------------------------------------- /utils/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sp-uhh/buddy/HEAD/utils/training_utils.py --------------------------------------------------------------------------------