├── LICENSE ├── README.md ├── configs ├── dataset │ ├── face_train.yaml │ ├── face_val.yaml │ ├── general_deg_codeformer_train.yaml │ ├── general_deg_codeformer_val.yaml │ ├── general_deg_realesrgan_train.yaml │ └── general_deg_realesrgan_val.yaml ├── model │ └── cldm.yaml └── train_cldm.yaml ├── dataset ├── batch_transform.py ├── codeformer.py ├── data_module.py └── realesrgan.py ├── figs ├── compare.png ├── logo.png └── model.png ├── inference.py ├── ldm ├── data │ ├── __init__.py │ └── util.py ├── models │ ├── __pycache__ │ │ └── autoencoder.cpython-39.pyc │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── ddim.cpython-39.pyc │ │ ├── ddpm.cpython-39.pyc │ │ └── ddpm.cpython-39.pyc-1 │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ ├── plms.py │ │ └── sampling_util.py ├── modules │ ├── __pycache__ │ │ ├── attention.cpython-39.pyc │ │ └── ema.cpython-39.pyc │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── model.cpython-39.pyc │ │ │ ├── openaimodel.cpython-39.pyc │ │ │ └── util.cpython-39.pyc │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── distributions.cpython-39.pyc │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── modules.cpython-39.pyc │ │ └── modules.py │ └── midas │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── blocks.py │ │ ├── dpt_depth.py │ │ ├── midas_net.py │ │ ├── midas_net_custom.py │ │ ├── transforms.py │ │ └── vit.py │ │ └── utils.py ├── util.py └── xformers_state.py ├── model ├── callbacks.py ├── cldm.py ├── cond_fn.py ├── mixins.py ├── spaced_sampler.py └── swinir.py ├── requirements.txt ├── scripts └── make_init_weight.py ├── train.py └── utils ├── common.py ├── degradation.py ├── face_restoration_helper.py ├── file.py ├── image ├── __init__.py ├── align_color.py ├── common.py ├── diffjpeg.py └── usm_sharp.py ├── metrics.py └── realesrgan ├── realesrganer.py └── rrdbnet.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/README.md -------------------------------------------------------------------------------- /configs/dataset/face_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/face_train.yaml -------------------------------------------------------------------------------- /configs/dataset/face_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/face_val.yaml -------------------------------------------------------------------------------- /configs/dataset/general_deg_codeformer_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/general_deg_codeformer_train.yaml -------------------------------------------------------------------------------- /configs/dataset/general_deg_codeformer_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/general_deg_codeformer_val.yaml -------------------------------------------------------------------------------- /configs/dataset/general_deg_realesrgan_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/general_deg_realesrgan_train.yaml -------------------------------------------------------------------------------- /configs/dataset/general_deg_realesrgan_val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/dataset/general_deg_realesrgan_val.yaml -------------------------------------------------------------------------------- /configs/model/cldm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/model/cldm.yaml -------------------------------------------------------------------------------- /configs/train_cldm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/configs/train_cldm.yaml -------------------------------------------------------------------------------- /dataset/batch_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/dataset/batch_transform.py -------------------------------------------------------------------------------- /dataset/codeformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/dataset/codeformer.py -------------------------------------------------------------------------------- /dataset/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/dataset/data_module.py -------------------------------------------------------------------------------- /dataset/realesrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/dataset/realesrgan.py -------------------------------------------------------------------------------- /figs/compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/figs/compare.png -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/figs/model.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/inference.py -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/data/util.py -------------------------------------------------------------------------------- /ldm/models/__pycache__/autoencoder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/__pycache__/autoencoder.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/models/diffusion/__pycache__/ddim.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/__pycache__/ddim.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/models/diffusion/__pycache__/ddpm.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/__pycache__/ddpm.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/models/diffusion/__pycache__/ddpm.cpython-39.pyc-1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/__pycache__/ddpm.cpython-39.pyc-1 -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /ldm/modules/__pycache__/attention.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/__pycache__/attention.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/__pycache__/ema.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/__pycache__/ema.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/distributions/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/distributions/__pycache__/distributions.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/distributions/__pycache__/distributions.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/encoders/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/encoders/__pycache__/modules.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/encoders/__pycache__/modules.cpython-39.pyc -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/api.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /ldm/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/modules/midas/utils.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/util.py -------------------------------------------------------------------------------- /ldm/xformers_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/ldm/xformers_state.py -------------------------------------------------------------------------------- /model/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/callbacks.py -------------------------------------------------------------------------------- /model/cldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/cldm.py -------------------------------------------------------------------------------- /model/cond_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/cond_fn.py -------------------------------------------------------------------------------- /model/mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/mixins.py -------------------------------------------------------------------------------- /model/spaced_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/spaced_sampler.py -------------------------------------------------------------------------------- /model/swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/model/swinir.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/make_init_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/scripts/make_init_weight.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/train.py -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/degradation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/degradation.py -------------------------------------------------------------------------------- /utils/face_restoration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/face_restoration_helper.py -------------------------------------------------------------------------------- /utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/file.py -------------------------------------------------------------------------------- /utils/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/image/__init__.py -------------------------------------------------------------------------------- /utils/image/align_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/image/align_color.py -------------------------------------------------------------------------------- /utils/image/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/image/common.py -------------------------------------------------------------------------------- /utils/image/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/image/diffjpeg.py -------------------------------------------------------------------------------- /utils/image/usm_sharp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/image/usm_sharp.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/realesrgan/realesrganer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/realesrgan/realesrganer.py -------------------------------------------------------------------------------- /utils/realesrgan/rrdbnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HVision-NKU/ControlSR/HEAD/utils/realesrgan/rrdbnet.py --------------------------------------------------------------------------------