From bd245f40eba42400283aad611e03559906411d7d Mon Sep 17 00:00:00 2001 From: csslc <22042099r@connect.polyu.hk> Date: Thu, 4 Jan 2024 21:45:50 +0800 Subject: [PATCH] release_code --- .idea/vcs.xml | 1 + .idea/workspace.xml | 11 +- README.md | 68 +- cal_iqa/cal_iqa.py | 236 ++ cal_iqa/iqa_G-STD.py | 111 + cal_iqa/iqa_L-STD.py | 190 ++ ...general_deg_stablesr_realesrgan_train.yaml | 63 + .../general_deg_stablesr_realesrgan_val.yaml | 64 + configs/model/ccsr_stage1.yaml | 108 + configs/model/ccsr_stage2.yaml | 118 + configs/train_ccsr_stage1.yaml | 47 + configs/train_ccsr_stage2.yaml | 49 + .../batch_transform.cpython-310.pyc | Bin 0 -> 8226 bytes .../batch_transform.cpython-37.pyc | Bin 0 -> 12024 bytes .../bicubic_torchvision.cpython-310.pyc | Bin 0 -> 2275 bytes .../__pycache__/data_module.cpython-310.pyc | Bin 0 -> 2727 bytes .../__pycache__/data_module.cpython-37.pyc | Bin 0 -> 2661 bytes .../__pycache__/projectdata.cpython-310.pyc | Bin 0 -> 3722 bytes .../__pycache__/projectdata.cpython-37.pyc | Bin 0 -> 3669 bytes .../__pycache__/realesrgan.cpython-310.pyc | Bin 0 -> 4046 bytes dataset/__pycache__/realesrgan.cpython-37.pyc | Bin 0 -> 3978 bytes dataset/batch_transform.py | 374 +++ dataset/bicubic_torchvision.py | 107 + dataset/codeformer.py | 109 + dataset/data_module.py | 68 + dataset/projectdata.py | 121 + dataset/realesrgan.py | 183 ++ figs/logo.png | Bin 18213 -> 30467 bytes inference_ccsr.py | 239 ++ ldm/__pycache__/util.cpython-310.pyc | Bin 0 -> 6165 bytes ldm/__pycache__/util.cpython-37.pyc | Bin 0 -> 6140 bytes .../xformers_state.cpython-310.pyc | Bin 0 -> 982 bytes ldm/__pycache__/xformers_state.cpython-37.pyc | Bin 0 -> 996 bytes ldm/data/__init__.py | 0 ldm/data/util.py | 24 + .../__pycache__/autoencoder.cpython-310.pyc | Bin 0 -> 7866 bytes .../__pycache__/autoencoder.cpython-37.pyc | Bin 0 -> 7766 bytes ldm/models/autoencoder.py | 224 ++ ldm/models/diffusion/__init__.py | 0 .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 173 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 170 bytes .../__pycache__/ddim.cpython-310.pyc | Bin 0 -> 9347 bytes .../diffusion/__pycache__/ddim.cpython-37.pyc | Bin 0 -> 9245 bytes .../__pycache__/ddpm.cpython-310.pyc | Bin 0 -> 53110 bytes .../diffusion/__pycache__/ddpm.cpython-37.pyc | Bin 0 -> 54432 bytes .../ddpm_ccsr_stage1.cpython-310.pyc | Bin 0 -> 54491 bytes .../ddpm_ccsr_stage2.cpython-310.pyc | Bin 0 -> 58092 bytes .../__pycache__/ddpm_psample2.cpython-310.pyc | Bin 0 -> 58248 bytes .../__pycache__/ddpm_psample3.cpython-310.pyc | Bin 0 -> 58248 bytes .../__pycache__/ddpm_qsample.cpython-310.pyc | Bin 0 -> 54428 bytes .../__pycache__/ddpm_qsample.cpython-37.pyc | Bin 0 -> 55828 bytes .../__pycache__/ddpm_qsample2.cpython-310.pyc | Bin 0 -> 54422 bytes .../__pycache__/ddpm_qsample2.cpython-37.pyc | Bin 0 -> 55829 bytes .../ddpm_qsample_0_8.cpython-310.pyc | Bin 0 -> 54425 bytes .../ddpm_qsample_0_8.cpython-37.pyc | Bin 0 -> 55832 bytes ..._qsample_finetunedecoder24.cpython-310.pyc | Bin 0 -> 57902 bytes ..._qsample_finetunedecoder30.cpython-310.pyc | Bin 0 -> 58014 bytes ..._qsample_finetunedecoder45.cpython-310.pyc | Bin 0 -> 58014 bytes ...m_qsample_finetunedecoder45.cpython-37.pyc | Bin 0 -> 59692 bytes ...le_finetunedecoder45_22_15.cpython-310.pyc | Bin 0 -> 58025 bytes ...ple_finetunedecoder45_22_15.cpython-37.pyc | Bin 0 -> 59703 bytes ...le_finetunedecoder45_30_22.cpython-310.pyc | Bin 0 -> 58025 bytes ...ple_finetunedecoder45_30_22.cpython-37.pyc | Bin 0 -> 59703 bytes ...ple_finetunedecoder45_30_8.cpython-310.pyc | Bin 0 -> 58037 bytes ...mple_finetunedecoder45_30_8.cpython-37.pyc | Bin 0 -> 59715 bytes ...le_finetunedecoder45_36_15.cpython-310.pyc | Bin 0 -> 58025 bytes ...ple_finetunedecoder45_36_15.cpython-37.pyc | Bin 0 -> 59703 bytes ...ample_finetunedecoder45_lr.cpython-310.pyc | Bin 0 -> 58088 bytes .../ddpm_qsample_lr.cpython-310.pyc | Bin 0 -> 54488 bytes ...fusion_finetune_x0_qsample.cpython-310.pyc | Bin 0 -> 88062 bytes ldm/models/diffusion/ddim.py | 336 +++ ldm/models/diffusion/ddpm.py | 1811 +++++++++++++++ ldm/models/diffusion/ddpm_ccsr_stage1.py | 1858 +++++++++++++++ ldm/models/diffusion/ddpm_ccsr_stage2.py | 2058 +++++++++++++++++ ldm/models/diffusion/dpm_solver/__init__.py | 1 + ldm/models/diffusion/dpm_solver/dpm_solver.py | 1154 +++++++++ ldm/models/diffusion/dpm_solver/sampler.py | 87 + ldm/models/diffusion/plms.py | 244 ++ ldm/models/diffusion/sampling_util.py | 22 + .../__pycache__/attention.cpython-310.pyc | Bin 0 -> 10381 bytes .../__pycache__/attention.cpython-37.pyc | Bin 0 -> 10534 bytes ldm/modules/__pycache__/ema.cpython-310.pyc | Bin 0 -> 3212 bytes ldm/modules/__pycache__/ema.cpython-37.pyc | Bin 0 -> 3186 bytes ldm/modules/attention.py | 343 +++ ldm/modules/diffusionmodules/__init__.py | 0 .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 181 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 178 bytes .../__pycache__/model.cpython-310.pyc | Bin 0 -> 20783 bytes .../__pycache__/model.cpython-37.pyc | Bin 0 -> 21736 bytes .../__pycache__/openaimodel.cpython-310.pyc | Bin 0 -> 20651 bytes .../__pycache__/openaimodel.cpython-37.pyc | Bin 0 -> 20785 bytes .../__pycache__/util.cpython-310.pyc | Bin 0 -> 9710 bytes .../__pycache__/util.cpython-37.pyc | Bin 0 -> 9626 bytes ldm/modules/diffusionmodules/model.py | 855 +++++++ ldm/modules/diffusionmodules/openaimodel.py | 786 +++++++ ldm/modules/diffusionmodules/upscaling.py | 81 + ldm/modules/diffusionmodules/util.py | 270 +++ ldm/modules/distributions/__init__.py | 0 .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 178 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 175 bytes .../__pycache__/distributions.cpython-310.pyc | Bin 0 -> 3776 bytes .../__pycache__/distributions.cpython-37.pyc | Bin 0 -> 3732 bytes ldm/modules/distributions/distributions.py | 92 + ldm/modules/ema.py | 80 + ldm/modules/encoders/__init__.py | 0 .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 173 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 170 bytes .../__pycache__/modules.cpython-310.pyc | Bin 0 -> 8440 bytes .../__pycache__/modules.cpython-37.pyc | Bin 0 -> 8516 bytes ldm/modules/encoders/modules.py | 215 ++ ldm/modules/losses/__init__.py | 1 + .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 251 bytes .../__pycache__/__init__.cpython-37.pyc | Bin 0 -> 248 bytes .../__pycache__/__init__.cpython-38.pyc | Bin 0 -> 258 bytes .../contperceptual.cpython-310.pyc | Bin 0 -> 4174 bytes .../__pycache__/contperceptual.cpython-37.pyc | Bin 0 -> 4166 bytes .../__pycache__/contperceptual.cpython-38.pyc | Bin 0 -> 4235 bytes ldm/modules/losses/contperceptual.py | 152 ++ ldm/modules/losses/vqperceptual.py | 180 ++ ldm/modules/midas/__init__.py | 0 ldm/modules/midas/api.py | 170 ++ ldm/modules/midas/midas/__init__.py | 0 ldm/modules/midas/midas/base_model.py | 16 + ldm/modules/midas/midas/blocks.py | 342 +++ ldm/modules/midas/midas/dpt_depth.py | 109 + ldm/modules/midas/midas/midas_net.py | 76 + ldm/modules/midas/midas/midas_net_custom.py | 128 + ldm/modules/midas/midas/transforms.py | 234 ++ ldm/modules/midas/midas/vit.py | 491 ++++ ldm/modules/midas/utils.py | 189 ++ ldm/util.py | 198 ++ ldm/xformers_state.py | 30 + model/__pycache__/callbacks.cpython-310.pyc | Bin 0 -> 2674 bytes model/__pycache__/callbacks.cpython-37.pyc | Bin 0 -> 2593 bytes model/__pycache__/ccsr_stage1.cpython-310.pyc | Bin 0 -> 10337 bytes model/__pycache__/ccsr_stage2.cpython-310.pyc | Bin 0 -> 10484 bytes model/__pycache__/cldm.cpython-310.pyc | Bin 0 -> 10352 bytes model/__pycache__/cldm.cpython-37.pyc | Bin 0 -> 10149 bytes .../__pycache__/cldm_decoder.cpython-310.pyc | Bin 0 -> 10502 bytes model/__pycache__/cldm_decoder.cpython-37.pyc | Bin 0 -> 10306 bytes .../cldm_decoder_24.cpython-310.pyc | Bin 0 -> 10505 bytes .../cldm_decoder_30.cpython-310.pyc | Bin 0 -> 10505 bytes .../cldm_decoder_45_22_15.cpython-310.pyc | Bin 0 -> 10517 bytes .../cldm_decoder_45_22_15.cpython-37.pyc | Bin 0 -> 10321 bytes .../cldm_decoder_45_30_22.cpython-310.pyc | Bin 0 -> 10517 bytes .../cldm_decoder_45_30_22.cpython-37.pyc | Bin 0 -> 10321 bytes .../cldm_decoder_45_30_8.cpython-310.pyc | Bin 0 -> 10515 bytes .../cldm_decoder_45_30_8.cpython-37.pyc | Bin 0 -> 10319 bytes .../cldm_decoder_45_36_15.cpython-310.pyc | Bin 0 -> 10517 bytes .../cldm_decoder_45_36_15.cpython-37.pyc | Bin 0 -> 10321 bytes .../cldm_decoder_lr.cpython-310.pyc | Bin 0 -> 10508 bytes model/__pycache__/cldm_lr.cpython-310.pyc | Bin 0 -> 10270 bytes .../cldm_lr_psample2.cpython-310.pyc | Bin 0 -> 10342 bytes .../cldm_lr_psample3.cpython-310.pyc | Bin 0 -> 10342 bytes .../cldm_lr_qsample.cpython-310.pyc | Bin 0 -> 10347 bytes .../cldm_lr_qsample.cpython-37.pyc | Bin 0 -> 10144 bytes .../cldm_lr_qsample2.cpython-310.pyc | Bin 0 -> 10342 bytes .../cldm_lr_qsample2.cpython-37.pyc | Bin 0 -> 10146 bytes .../cldm_lr_qsample_0_8.cpython-310.pyc | Bin 0 -> 10348 bytes .../cldm_lr_qsample_0_8.cpython-37.pyc | Bin 0 -> 10152 bytes .../cldm_lr_qsample_lr.cpython-310.pyc | Bin 0 -> 10346 bytes model/__pycache__/cond_fn.cpython-310.pyc | Bin 0 -> 2949 bytes model/__pycache__/cond_fn.cpython-37.pyc | Bin 0 -> 2913 bytes model/__pycache__/mixins.cpython-310.pyc | Bin 0 -> 640 bytes model/__pycache__/mixins.cpython-37.pyc | Bin 0 -> 625 bytes model/__pycache__/p_sampler.cpython-310.pyc | Bin 0 -> 16149 bytes model/__pycache__/q_sampler.cpython-310.pyc | Bin 0 -> 21269 bytes model/__pycache__/q_sampler.cpython-37.pyc | Bin 0 -> 32803 bytes .../spaced_sampler.cpython-310.pyc | Bin 0 -> 16422 bytes .../__pycache__/spaced_sampler.cpython-37.pyc | Bin 0 -> 16581 bytes model/__pycache__/swinir.cpython-310.pyc | Bin 0 -> 31251 bytes model/callbacks.py | 74 + model/ccsr_stage1.py | 406 ++++ model/ccsr_stage2.py | 407 ++++ model/cond_fn.py | 78 + model/mixins.py | 9 + model/q_sampler.py | 1043 +++++++++ model/spaced_sampler.py | 670 ++++++ preset/test_datasets/19.jpg | Bin 0 -> 13552 bytes preset/test_datasets/49.jpg | Bin 0 -> 2790 bytes requirements.txt | 19 + scripts/make_file_list.py | 36 + scripts/make_stage2_init_weight.py | 80 + scripts/test.sh | 0 sh/1_pip.sh | 3 + sh/2_inference.sh | 14 + sh/3_create_weight.sh | 11 + sh/4_generate_data_file.sh | 10 + sh/5_train_stage1.sh | 3 + sh/6_train_stage2.sh | 3 + train.py | 32 + utils/__pycache__/common.cpython-310.pyc | Bin 0 -> 2069 bytes utils/__pycache__/common.cpython-37.pyc | Bin 0 -> 2060 bytes utils/__pycache__/degradation.cpython-310.pyc | Bin 0 -> 20296 bytes utils/__pycache__/degradation.cpython-37.pyc | Bin 0 -> 22425 bytes utils/__pycache__/file.cpython-310.pyc | Bin 0 -> 2708 bytes utils/__pycache__/file.cpython-37.pyc | Bin 0 -> 2625 bytes utils/__pycache__/metrics.cpython-310.pyc | Bin 0 -> 2652 bytes utils/common.py | 51 + utils/degradation.py | 765 ++++++ utils/face_restoration_helper.py | 517 +++++ utils/file.py | 103 + utils/image/__init__.py | 26 + .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 591 bytes .../image/__pycache__/__init__.cpython-37.pyc | Bin 0 -> 624 bytes .../__pycache__/align_color.cpython-310.pyc | Bin 0 -> 3716 bytes .../__pycache__/align_color.cpython-37.pyc | Bin 0 -> 3843 bytes .../image/__pycache__/common.cpython-310.pyc | Bin 0 -> 7506 bytes utils/image/__pycache__/common.cpython-37.pyc | Bin 0 -> 7517 bytes .../__pycache__/diffjpeg.cpython-310.pyc | Bin 0 -> 14822 bytes .../image/__pycache__/diffjpeg.cpython-37.pyc | Bin 0 -> 16804 bytes .../__pycache__/usm_sharp.cpython-310.pyc | Bin 0 -> 1235 bytes .../__pycache__/usm_sharp.cpython-37.pyc | Bin 0 -> 1220 bytes utils/image/align_color.py | 119 + utils/image/common.py | 235 ++ utils/image/diffjpeg.py | 492 ++++ utils/image/usm_sharp.py | 29 + utils/metrics.py | 66 + utils/realesrgan/realesrganer.py | 339 +++ utils/realesrgan/rrdbnet.py | 183 ++ 220 files changed, 20910 insertions(+), 37 deletions(-) create mode 100644 cal_iqa/cal_iqa.py create mode 100644 cal_iqa/iqa_G-STD.py create mode 100644 cal_iqa/iqa_L-STD.py create mode 100644 configs/dataset/general_deg_stablesr_realesrgan_train.yaml create mode 100644 configs/dataset/general_deg_stablesr_realesrgan_val.yaml create mode 100644 configs/model/ccsr_stage1.yaml create mode 100644 configs/model/ccsr_stage2.yaml create mode 100644 configs/train_ccsr_stage1.yaml create mode 100644 configs/train_ccsr_stage2.yaml create mode 100644 dataset/__pycache__/batch_transform.cpython-310.pyc create mode 100644 dataset/__pycache__/batch_transform.cpython-37.pyc create mode 100644 dataset/__pycache__/bicubic_torchvision.cpython-310.pyc create mode 100644 dataset/__pycache__/data_module.cpython-310.pyc create mode 100644 dataset/__pycache__/data_module.cpython-37.pyc create mode 100644 dataset/__pycache__/projectdata.cpython-310.pyc create mode 100644 dataset/__pycache__/projectdata.cpython-37.pyc create mode 100644 dataset/__pycache__/realesrgan.cpython-310.pyc create mode 100644 dataset/__pycache__/realesrgan.cpython-37.pyc create mode 100644 dataset/batch_transform.py create mode 100644 dataset/bicubic_torchvision.py create mode 100644 dataset/codeformer.py create mode 100644 dataset/data_module.py create mode 100644 dataset/projectdata.py create mode 100644 dataset/realesrgan.py create mode 100644 inference_ccsr.py create mode 100644 ldm/__pycache__/util.cpython-310.pyc create mode 100644 ldm/__pycache__/util.cpython-37.pyc create mode 100644 ldm/__pycache__/xformers_state.cpython-310.pyc create mode 100644 ldm/__pycache__/xformers_state.cpython-37.pyc create mode 100644 ldm/data/__init__.py create mode 100644 ldm/data/util.py create mode 100644 ldm/models/__pycache__/autoencoder.cpython-310.pyc create mode 100644 ldm/models/__pycache__/autoencoder.cpython-37.pyc create mode 100644 ldm/models/autoencoder.py create mode 100644 ldm/models/diffusion/__init__.py create mode 100644 ldm/models/diffusion/__pycache__/__init__.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/__init__.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddim.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddim.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_ccsr_stage1.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_ccsr_stage2.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_psample2.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_psample3.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample2.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample2.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_0_8.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_0_8.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder24.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder30.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_22_15.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_22_15.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_30_22.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_30_22.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_30_8.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_30_8.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_36_15.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_36_15.cpython-37.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_finetunedecoder45_lr.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_qsample_lr.cpython-310.pyc create mode 100644 ldm/models/diffusion/__pycache__/ddpm_truncted_diffusion_finetune_x0_qsample.cpython-310.pyc create mode 100644 ldm/models/diffusion/ddim.py create mode 100644 ldm/models/diffusion/ddpm.py create mode 100644 ldm/models/diffusion/ddpm_ccsr_stage1.py create mode 100644 ldm/models/diffusion/ddpm_ccsr_stage2.py create mode 100644 ldm/models/diffusion/dpm_solver/__init__.py create mode 100644 ldm/models/diffusion/dpm_solver/dpm_solver.py create mode 100644 ldm/models/diffusion/dpm_solver/sampler.py create mode 100644 ldm/models/diffusion/plms.py create mode 100644 ldm/models/diffusion/sampling_util.py create mode 100644 ldm/modules/__pycache__/attention.cpython-310.pyc create mode 100644 ldm/modules/__pycache__/attention.cpython-37.pyc create mode 100644 ldm/modules/__pycache__/ema.cpython-310.pyc create mode 100644 ldm/modules/__pycache__/ema.cpython-37.pyc create mode 100644 ldm/modules/attention.py create mode 100644 ldm/modules/diffusionmodules/__init__.py create mode 100644 ldm/modules/diffusionmodules/__pycache__/__init__.cpython-310.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/__init__.cpython-37.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/model.cpython-310.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/model.cpython-37.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-310.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-37.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/util.cpython-310.pyc create mode 100644 ldm/modules/diffusionmodules/__pycache__/util.cpython-37.pyc create mode 100644 ldm/modules/diffusionmodules/model.py create mode 100644 ldm/modules/diffusionmodules/openaimodel.py create mode 100644 ldm/modules/diffusionmodules/upscaling.py create mode 100644 ldm/modules/diffusionmodules/util.py create mode 100644 ldm/modules/distributions/__init__.py create mode 100644 ldm/modules/distributions/__pycache__/__init__.cpython-310.pyc create mode 100644 ldm/modules/distributions/__pycache__/__init__.cpython-37.pyc create mode 100644 ldm/modules/distributions/__pycache__/distributions.cpython-310.pyc create mode 100644 ldm/modules/distributions/__pycache__/distributions.cpython-37.pyc create mode 100644 ldm/modules/distributions/distributions.py create mode 100644 ldm/modules/ema.py create mode 100644 ldm/modules/encoders/__init__.py create mode 100644 ldm/modules/encoders/__pycache__/__init__.cpython-310.pyc create mode 100644 ldm/modules/encoders/__pycache__/__init__.cpython-37.pyc create mode 100644 ldm/modules/encoders/__pycache__/modules.cpython-310.pyc create mode 100644 ldm/modules/encoders/__pycache__/modules.cpython-37.pyc create mode 100644 ldm/modules/encoders/modules.py create mode 100644 ldm/modules/losses/__init__.py create mode 100644 ldm/modules/losses/__pycache__/__init__.cpython-310.pyc create mode 100644 ldm/modules/losses/__pycache__/__init__.cpython-37.pyc create mode 100644 ldm/modules/losses/__pycache__/__init__.cpython-38.pyc create mode 100644 ldm/modules/losses/__pycache__/contperceptual.cpython-310.pyc create mode 100644 ldm/modules/losses/__pycache__/contperceptual.cpython-37.pyc create mode 100644 ldm/modules/losses/__pycache__/contperceptual.cpython-38.pyc create mode 100644 ldm/modules/losses/contperceptual.py create mode 100644 ldm/modules/losses/vqperceptual.py create mode 100644 ldm/modules/midas/__init__.py create mode 100644 ldm/modules/midas/api.py create mode 100644 ldm/modules/midas/midas/__init__.py create mode 100644 ldm/modules/midas/midas/base_model.py create mode 100644 ldm/modules/midas/midas/blocks.py create mode 100644 ldm/modules/midas/midas/dpt_depth.py create mode 100644 ldm/modules/midas/midas/midas_net.py create mode 100644 ldm/modules/midas/midas/midas_net_custom.py create mode 100644 ldm/modules/midas/midas/transforms.py create mode 100644 ldm/modules/midas/midas/vit.py create mode 100644 ldm/modules/midas/utils.py create mode 100644 ldm/util.py create mode 100644 ldm/xformers_state.py create mode 100644 model/__pycache__/callbacks.cpython-310.pyc create mode 100644 model/__pycache__/callbacks.cpython-37.pyc create mode 100644 model/__pycache__/ccsr_stage1.cpython-310.pyc create mode 100644 model/__pycache__/ccsr_stage2.cpython-310.pyc create mode 100644 model/__pycache__/cldm.cpython-310.pyc create mode 100644 model/__pycache__/cldm.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder_24.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_30.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_45_22_15.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_45_22_15.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder_45_30_22.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_45_30_22.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder_45_30_8.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_45_30_8.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder_45_36_15.cpython-310.pyc create mode 100644 model/__pycache__/cldm_decoder_45_36_15.cpython-37.pyc create mode 100644 model/__pycache__/cldm_decoder_lr.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_psample2.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_psample3.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_qsample.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_qsample.cpython-37.pyc create mode 100644 model/__pycache__/cldm_lr_qsample2.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_qsample2.cpython-37.pyc create mode 100644 model/__pycache__/cldm_lr_qsample_0_8.cpython-310.pyc create mode 100644 model/__pycache__/cldm_lr_qsample_0_8.cpython-37.pyc create mode 100644 model/__pycache__/cldm_lr_qsample_lr.cpython-310.pyc create mode 100644 model/__pycache__/cond_fn.cpython-310.pyc create mode 100644 model/__pycache__/cond_fn.cpython-37.pyc create mode 100644 model/__pycache__/mixins.cpython-310.pyc create mode 100644 model/__pycache__/mixins.cpython-37.pyc create mode 100644 model/__pycache__/p_sampler.cpython-310.pyc create mode 100644 model/__pycache__/q_sampler.cpython-310.pyc create mode 100644 model/__pycache__/q_sampler.cpython-37.pyc create mode 100644 model/__pycache__/spaced_sampler.cpython-310.pyc create mode 100644 model/__pycache__/spaced_sampler.cpython-37.pyc create mode 100644 model/__pycache__/swinir.cpython-310.pyc create mode 100644 model/callbacks.py create mode 100644 model/ccsr_stage1.py create mode 100644 model/ccsr_stage2.py create mode 100644 model/cond_fn.py create mode 100644 model/mixins.py create mode 100644 model/q_sampler.py create mode 100644 model/spaced_sampler.py create mode 100644 preset/test_datasets/19.jpg create mode 100644 preset/test_datasets/49.jpg create mode 100644 requirements.txt create mode 100644 scripts/make_file_list.py create mode 100644 scripts/make_stage2_init_weight.py create mode 100644 scripts/test.sh create mode 100644 sh/1_pip.sh create mode 100644 sh/2_inference.sh create mode 100644 sh/3_create_weight.sh create mode 100644 sh/4_generate_data_file.sh create mode 100644 sh/5_train_stage1.sh create mode 100644 sh/6_train_stage2.sh create mode 100644 train.py create mode 100644 utils/__pycache__/common.cpython-310.pyc create mode 100644 utils/__pycache__/common.cpython-37.pyc create mode 100644 utils/__pycache__/degradation.cpython-310.pyc create mode 100644 utils/__pycache__/degradation.cpython-37.pyc create mode 100644 utils/__pycache__/file.cpython-310.pyc create mode 100644 utils/__pycache__/file.cpython-37.pyc create mode 100644 utils/__pycache__/metrics.cpython-310.pyc create mode 100644 utils/common.py create mode 100644 utils/degradation.py create mode 100644 utils/face_restoration_helper.py create mode 100644 utils/file.py create mode 100644 utils/image/__init__.py create mode 100644 utils/image/__pycache__/__init__.cpython-310.pyc create mode 100644 utils/image/__pycache__/__init__.cpython-37.pyc create mode 100644 utils/image/__pycache__/align_color.cpython-310.pyc create mode 100644 utils/image/__pycache__/align_color.cpython-37.pyc create mode 100644 utils/image/__pycache__/common.cpython-310.pyc create mode 100644 utils/image/__pycache__/common.cpython-37.pyc create mode 100644 utils/image/__pycache__/diffjpeg.cpython-310.pyc create mode 100644 utils/image/__pycache__/diffjpeg.cpython-37.pyc create mode 100644 utils/image/__pycache__/usm_sharp.cpython-310.pyc create mode 100644 utils/image/__pycache__/usm_sharp.cpython-37.pyc create mode 100644 utils/image/align_color.py create mode 100644 utils/image/common.py create mode 100644 utils/image/diffjpeg.py create mode 100644 utils/image/usm_sharp.py create mode 100644 utils/metrics.py create mode 100644 utils/realesrgan/realesrganer.py create mode 100644 utils/realesrgan/rrdbnet.py diff --git a/.idea/vcs.xml b/.idea/vcs.xml index 94a25f7..f6eb092 100644 --- a/.idea/vcs.xml +++ b/.idea/vcs.xml @@ -2,5 +2,6 @@ + \ No newline at end of file diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 58203d5..e17ae08 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -14,16 +14,17 @@