├── .gitignore ├── LICENSE ├── README.md ├── app.py ├── config ├── denoiser.yaml ├── enhancer_stage1.yaml └── enhancer_stage2.yaml ├── packages.txt ├── pyproject.toml ├── requirements.txt ├── resemble_enhance ├── __init__.py ├── common.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── distorter │ │ ├── __init__.py │ │ ├── base.py │ │ ├── custom.py │ │ ├── distorter.py │ │ └── sox.py │ └── utils.py ├── denoiser │ ├── __init__.py │ ├── __main__.py │ ├── denoiser.py │ ├── hparams.py │ ├── inference.py │ ├── train.py │ └── unet.py ├── enhancer │ ├── __init__.py │ ├── __main__.py │ ├── download.py │ ├── enhancer.py │ ├── hparams.py │ ├── inference.py │ ├── lcfm │ │ ├── __init__.py │ │ ├── cfm.py │ │ ├── irmae.py │ │ ├── lcfm.py │ │ └── wn.py │ ├── train.py │ └── univnet │ │ ├── __init__.py │ │ ├── alias_free_torch │ │ ├── __init__.py │ │ ├── filter.py │ │ └── resample.py │ │ ├── amp.py │ │ ├── discriminator.py │ │ ├── lvcnet.py │ │ ├── mrstft.py │ │ └── univnet.py ├── hparams.py ├── inference.py ├── melspec.py └── utils │ ├── __init__.py │ ├── control.py │ ├── distributed.py │ ├── engine.py │ ├── logging.py │ ├── train_loop.py │ └── utils.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/app.py -------------------------------------------------------------------------------- /config/denoiser.yaml: -------------------------------------------------------------------------------- 1 | batch_size_per_gpu: 32 2 | training_seconds: 3.0 3 | -------------------------------------------------------------------------------- /config/enhancer_stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/config/enhancer_stage1.yaml -------------------------------------------------------------------------------- /config/enhancer_stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/config/enhancer_stage2.yaml -------------------------------------------------------------------------------- /packages.txt: -------------------------------------------------------------------------------- 1 | libsox-dev 2 | -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/requirements.txt -------------------------------------------------------------------------------- /resemble_enhance/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resemble_enhance/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/common.py -------------------------------------------------------------------------------- /resemble_enhance/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/__init__.py -------------------------------------------------------------------------------- /resemble_enhance/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/dataset.py -------------------------------------------------------------------------------- /resemble_enhance/data/distorter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/distorter/__init__.py -------------------------------------------------------------------------------- /resemble_enhance/data/distorter/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/distorter/base.py -------------------------------------------------------------------------------- /resemble_enhance/data/distorter/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/distorter/custom.py -------------------------------------------------------------------------------- /resemble_enhance/data/distorter/distorter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/distorter/distorter.py -------------------------------------------------------------------------------- /resemble_enhance/data/distorter/sox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/distorter/sox.py -------------------------------------------------------------------------------- /resemble_enhance/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/data/utils.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resemble_enhance/denoiser/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/__main__.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/denoiser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/denoiser.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/hparams.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/inference.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/train.py -------------------------------------------------------------------------------- /resemble_enhance/denoiser/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/denoiser/unet.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resemble_enhance/enhancer/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/__main__.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/download.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/enhancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/enhancer.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/hparams.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/inference.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/lcfm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/lcfm/__init__.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/lcfm/cfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/lcfm/cfm.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/lcfm/irmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/lcfm/irmae.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/lcfm/lcfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/lcfm/lcfm.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/lcfm/wn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/lcfm/wn.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/train.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/__init__.py: -------------------------------------------------------------------------------- 1 | from .univnet import UnivNet 2 | -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/alias_free_torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/alias_free_torch/__init__.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/alias_free_torch/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/alias_free_torch/filter.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/alias_free_torch/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/alias_free_torch/resample.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/amp.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/discriminator.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/lvcnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/lvcnet.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/mrstft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/mrstft.py -------------------------------------------------------------------------------- /resemble_enhance/enhancer/univnet/univnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/enhancer/univnet/univnet.py -------------------------------------------------------------------------------- /resemble_enhance/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/hparams.py -------------------------------------------------------------------------------- /resemble_enhance/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/inference.py -------------------------------------------------------------------------------- /resemble_enhance/melspec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/melspec.py -------------------------------------------------------------------------------- /resemble_enhance/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/__init__.py -------------------------------------------------------------------------------- /resemble_enhance/utils/control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/control.py -------------------------------------------------------------------------------- /resemble_enhance/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/distributed.py -------------------------------------------------------------------------------- /resemble_enhance/utils/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/engine.py -------------------------------------------------------------------------------- /resemble_enhance/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/logging.py -------------------------------------------------------------------------------- /resemble_enhance/utils/train_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/train_loop.py -------------------------------------------------------------------------------- /resemble_enhance/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/resemble_enhance/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/resemble-ai/resemble-enhance/HEAD/setup.py --------------------------------------------------------------------------------