├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── config ├── ablation_higheralpha_75s.yaml ├── ablation_higheralpha_75s_globsigy.yaml ├── ablation_nocqt_nowav_75s.yaml ├── baseline_regression_75s.yaml ├── baseline_scoredec_75s.yaml ├── callbacks │ ├── ema_last_and_every_50k_steps.yaml │ └── ema_last_sisdr_pesq.yaml ├── datamodule │ ├── README.md │ ├── defaults.yaml │ └── example.yaml ├── flowdec_25s.yaml ├── flowdec_25s_globsigy.yaml ├── flowdec_75m.yaml ├── flowdec_75m_globsigy.yaml ├── flowdec_75s.yaml ├── flowdec_75s_globsigy.yaml ├── global_defaults.yaml ├── logger │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ ├── all_model_defaults.yaml │ ├── backbone │ │ ├── ncsnpp_default_ycond.yaml │ │ └── ncsnpp_final_no_attn.yaml │ ├── eval_metrics │ │ ├── sisxr_logspec_mse.yaml │ │ └── visqol_sisxr_logspec_mse.yaml │ ├── eval_variants │ │ ├── 20k_steps_midpoint_6_25.yaml │ │ └── plain_forward.yaml │ ├── feature_extractor │ │ ├── compressed_complex_stft_final.yaml │ │ └── compressed_complex_stft_sgmse1534.yaml │ ├── flow_model_final.yaml │ ├── flow_model_sgmse.yaml │ ├── regression_model_final.yaml │ ├── score_model_final.yaml │ ├── score_model_sgmse.yaml │ └── sde │ │ ├── ouve_final.yaml │ │ └── ouve_sgmse.yaml └── optimizer │ └── adam.yaml ├── data ├── flowdec_autoparams_25s.npy └── flowdec_autoparams_75m.npy ├── demo.ipynb ├── dummy_paired_filelist.txt ├── enhance.py ├── flowdec ├── __init__.py ├── backbones │ ├── 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 ├── callbacks │ └── ema.py ├── data │ ├── data_module.py │ ├── feature_extractors.py │ └── sigma_models │ │ └── __init__.py ├── eval │ ├── metrics.py │ ├── sigmos │ │ ├── LICENSE │ │ ├── README.md │ │ ├── Transparency_FAQ.md │ │ └── sigmos.py │ └── visqol │ │ ├── LICENSE │ │ └── __init__.py ├── losses.py ├── model.py ├── sampling │ ├── __init__.py │ ├── correctors.py │ ├── predictors.py │ ├── solvers.py │ └── solvers_torchdiffeq.py ├── sdes.py └── util │ ├── hydra.py │ ├── logging.py │ ├── other.py │ ├── registry.py │ └── viz.py ├── requirements.txt ├── scripts └── estimate_flowdec_params.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/README.md -------------------------------------------------------------------------------- /config/ablation_higheralpha_75s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/ablation_higheralpha_75s.yaml -------------------------------------------------------------------------------- /config/ablation_higheralpha_75s_globsigy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/ablation_higheralpha_75s_globsigy.yaml -------------------------------------------------------------------------------- /config/ablation_nocqt_nowav_75s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/ablation_nocqt_nowav_75s.yaml -------------------------------------------------------------------------------- /config/baseline_regression_75s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/baseline_regression_75s.yaml -------------------------------------------------------------------------------- /config/baseline_scoredec_75s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/baseline_scoredec_75s.yaml -------------------------------------------------------------------------------- /config/callbacks/ema_last_and_every_50k_steps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/callbacks/ema_last_and_every_50k_steps.yaml -------------------------------------------------------------------------------- /config/callbacks/ema_last_sisdr_pesq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/callbacks/ema_last_sisdr_pesq.yaml -------------------------------------------------------------------------------- /config/datamodule/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/datamodule/README.md -------------------------------------------------------------------------------- /config/datamodule/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/datamodule/defaults.yaml -------------------------------------------------------------------------------- /config/datamodule/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/datamodule/example.yaml -------------------------------------------------------------------------------- /config/flowdec_25s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_25s.yaml -------------------------------------------------------------------------------- /config/flowdec_25s_globsigy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_25s_globsigy.yaml -------------------------------------------------------------------------------- /config/flowdec_75m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_75m.yaml -------------------------------------------------------------------------------- /config/flowdec_75m_globsigy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_75m_globsigy.yaml -------------------------------------------------------------------------------- /config/flowdec_75s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_75s.yaml -------------------------------------------------------------------------------- /config/flowdec_75s_globsigy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/flowdec_75s_globsigy.yaml -------------------------------------------------------------------------------- /config/global_defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/global_defaults.yaml -------------------------------------------------------------------------------- /config/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/logger/tensorboard.yaml -------------------------------------------------------------------------------- /config/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/logger/wandb.yaml -------------------------------------------------------------------------------- /config/model/all_model_defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/all_model_defaults.yaml -------------------------------------------------------------------------------- /config/model/backbone/ncsnpp_default_ycond.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/backbone/ncsnpp_default_ycond.yaml -------------------------------------------------------------------------------- /config/model/backbone/ncsnpp_final_no_attn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/backbone/ncsnpp_final_no_attn.yaml -------------------------------------------------------------------------------- /config/model/eval_metrics/sisxr_logspec_mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/eval_metrics/sisxr_logspec_mse.yaml -------------------------------------------------------------------------------- /config/model/eval_metrics/visqol_sisxr_logspec_mse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/eval_metrics/visqol_sisxr_logspec_mse.yaml -------------------------------------------------------------------------------- /config/model/eval_variants/20k_steps_midpoint_6_25.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/eval_variants/20k_steps_midpoint_6_25.yaml -------------------------------------------------------------------------------- /config/model/eval_variants/plain_forward.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/eval_variants/plain_forward.yaml -------------------------------------------------------------------------------- /config/model/feature_extractor/compressed_complex_stft_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/feature_extractor/compressed_complex_stft_final.yaml -------------------------------------------------------------------------------- /config/model/feature_extractor/compressed_complex_stft_sgmse1534.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/feature_extractor/compressed_complex_stft_sgmse1534.yaml -------------------------------------------------------------------------------- /config/model/flow_model_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/flow_model_final.yaml -------------------------------------------------------------------------------- /config/model/flow_model_sgmse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/flow_model_sgmse.yaml -------------------------------------------------------------------------------- /config/model/regression_model_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/regression_model_final.yaml -------------------------------------------------------------------------------- /config/model/score_model_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/score_model_final.yaml -------------------------------------------------------------------------------- /config/model/score_model_sgmse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/score_model_sgmse.yaml -------------------------------------------------------------------------------- /config/model/sde/ouve_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/sde/ouve_final.yaml -------------------------------------------------------------------------------- /config/model/sde/ouve_sgmse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/model/sde/ouve_sgmse.yaml -------------------------------------------------------------------------------- /config/optimizer/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/config/optimizer/adam.yaml -------------------------------------------------------------------------------- /data/flowdec_autoparams_25s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/data/flowdec_autoparams_25s.npy -------------------------------------------------------------------------------- /data/flowdec_autoparams_75m.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/data/flowdec_autoparams_75m.npy -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/demo.ipynb -------------------------------------------------------------------------------- /dummy_paired_filelist.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/dummy_paired_filelist.txt -------------------------------------------------------------------------------- /enhance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/enhance.py -------------------------------------------------------------------------------- /flowdec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/__init__.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/layers.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/layerspp.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/normalization.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/__init__.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/fused_act.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/upfirdn2d.py -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /flowdec/backbones/ncsnpp_utils/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/backbones/ncsnpp_utils/up_or_down_sampling.py -------------------------------------------------------------------------------- /flowdec/callbacks/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/callbacks/ema.py -------------------------------------------------------------------------------- /flowdec/data/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/data/data_module.py -------------------------------------------------------------------------------- /flowdec/data/feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/data/feature_extractors.py -------------------------------------------------------------------------------- /flowdec/data/sigma_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/data/sigma_models/__init__.py -------------------------------------------------------------------------------- /flowdec/eval/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/metrics.py -------------------------------------------------------------------------------- /flowdec/eval/sigmos/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/sigmos/LICENSE -------------------------------------------------------------------------------- /flowdec/eval/sigmos/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/sigmos/README.md -------------------------------------------------------------------------------- /flowdec/eval/sigmos/Transparency_FAQ.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/sigmos/Transparency_FAQ.md -------------------------------------------------------------------------------- /flowdec/eval/sigmos/sigmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/sigmos/sigmos.py -------------------------------------------------------------------------------- /flowdec/eval/visqol/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/visqol/LICENSE -------------------------------------------------------------------------------- /flowdec/eval/visqol/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/eval/visqol/__init__.py -------------------------------------------------------------------------------- /flowdec/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/losses.py -------------------------------------------------------------------------------- /flowdec/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/model.py -------------------------------------------------------------------------------- /flowdec/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sampling/__init__.py -------------------------------------------------------------------------------- /flowdec/sampling/correctors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sampling/correctors.py -------------------------------------------------------------------------------- /flowdec/sampling/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sampling/predictors.py -------------------------------------------------------------------------------- /flowdec/sampling/solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sampling/solvers.py -------------------------------------------------------------------------------- /flowdec/sampling/solvers_torchdiffeq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sampling/solvers_torchdiffeq.py -------------------------------------------------------------------------------- /flowdec/sdes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/sdes.py -------------------------------------------------------------------------------- /flowdec/util/hydra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/util/hydra.py -------------------------------------------------------------------------------- /flowdec/util/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/util/logging.py -------------------------------------------------------------------------------- /flowdec/util/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/util/other.py -------------------------------------------------------------------------------- /flowdec/util/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/util/registry.py -------------------------------------------------------------------------------- /flowdec/util/viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/flowdec/util/viz.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/estimate_flowdec_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/scripts/estimate_flowdec_params.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/FlowDec/HEAD/train.py --------------------------------------------------------------------------------