├── .idea ├── CCSR.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── modules.xml ├── vcs.xml └── workspace.xml ├── ADD ├── dnnlib │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ └── util.cpython-310.pyc │ └── util.py ├── layers │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ ├── attention.cpython-310.pyc │ │ ├── block.cpython-310.pyc │ │ ├── dino_head.cpython-310.pyc │ │ ├── drop_path.cpython-310.pyc │ │ ├── layer_scale.cpython-310.pyc │ │ ├── mlp.cpython-310.pyc │ │ ├── patch_embed.cpython-310.pyc │ │ └── swiglu_ffn.cpython-310.pyc │ ├── attention.py │ ├── block.py │ ├── dino_head.py │ ├── drop_path.py │ ├── layer_scale.py │ ├── mlp.py │ ├── patch_embed.py │ └── swiglu_ffn.py ├── models │ ├── __pycache__ │ │ ├── discriminator.cpython-310.pyc │ │ └── vit.cpython-310.pyc │ ├── discriminator.py │ └── vit.py ├── th_utils │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ ├── custom_ops.cpython-310.pyc │ │ └── misc.cpython-310.pyc │ ├── custom_ops.py │ ├── misc.py │ └── ops │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ └── bias_act.cpython-310.pyc │ │ ├── bias_act.cpp │ │ ├── bias_act.cu │ │ ├── bias_act.h │ │ ├── bias_act.py │ │ ├── conv2d_gradfix.py │ │ ├── conv2d_resample.py │ │ ├── filtered_lrelu.cpp │ │ ├── filtered_lrelu.cu │ │ ├── filtered_lrelu.h │ │ ├── filtered_lrelu.py │ │ ├── filtered_lrelu_ns.cu │ │ ├── filtered_lrelu_rd.cu │ │ ├── filtered_lrelu_wr.cu │ │ ├── fma.py │ │ ├── grid_sample_gradfix.py │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.cu │ │ ├── upfirdn2d.h │ │ └── upfirdn2d.py └── utils │ ├── __pycache__ │ └── util_net.cpython-310.pyc │ └── util_net.py ├── LICENSE ├── README.md ├── dataloaders ├── __pycache__ │ ├── paired_dataset_txt.cpython-310.pyc │ └── realesrgan.cpython-310.pyc ├── paired_dataset_txt.py ├── params_ccsr.yml └── realesrgan.py ├── figs ├── compare_1.png ├── compare_2.png ├── compare_3.png ├── compare_4.png ├── compare_efficient.png ├── compare_standard.png ├── fig.png ├── framework.png ├── logo.png └── table.png ├── models ├── DiffAugment.py ├── __pycache__ │ ├── DiffAugment.cpython-310.pyc │ ├── controlnet.cpython-310.pyc │ ├── shared.cpython-310.pyc │ ├── unet_2d_blocks.cpython-310.pyc │ ├── unet_2d_condition.cpython-310.pyc │ └── vit_utils.cpython-310.pyc ├── controlnet.py ├── losses │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-310.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── contperceptual.cpython-310.pyc │ │ ├── contperceptual.cpython-37.pyc │ │ └── contperceptual.cpython-38.pyc │ ├── contperceptual.py │ └── vqperceptual.py ├── shared.py ├── unet_2d_blocks.py ├── unet_2d_condition.py └── vit_utils.py ├── myutils ├── __pycache__ │ ├── devices.cpython-310.pyc │ └── wavelet_color_fix.cpython-310.pyc ├── devices.py ├── img_util.py ├── misc.py ├── vaehook.py └── wavelet_color_fix.py ├── pipelines ├── __pycache__ │ └── pipeline_ccsr.cpython-310.pyc └── pipeline_ccsr.py ├── requirements.txt ├── scripts ├── get_path.py ├── test │ ├── test_ccsr_multistep.sh │ ├── test_ccsr_onestep.sh │ └── test_ccsr_tile.sh └── train │ ├── train_ccsr_stage1.sh │ ├── train_ccsr_stage2.sh │ └── train_controlnet.sh ├── test_ccsr_tile.py ├── train_ccsr_stage1.py ├── train_ccsr_stage2.py ├── train_controlnet.py └── utils ├── __pycache__ └── vaehook.cpython-310.pyc ├── devices.py ├── img_util.py ├── misc.py ├── vaehook.py └── wavelet_color_fix.py /.idea/CCSR.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/CCSR.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/.idea/workspace.xml -------------------------------------------------------------------------------- /ADD/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/dnnlib/__init__.py -------------------------------------------------------------------------------- /ADD/dnnlib/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/dnnlib/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/dnnlib/__pycache__/util.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/dnnlib/__pycache__/util.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/dnnlib/util.py -------------------------------------------------------------------------------- /ADD/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__init__.py -------------------------------------------------------------------------------- /ADD/layers/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/attention.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/attention.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/block.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/block.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/dino_head.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/dino_head.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/drop_path.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/drop_path.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/layer_scale.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/layer_scale.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/mlp.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/mlp.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/patch_embed.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/patch_embed.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/__pycache__/swiglu_ffn.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/__pycache__/swiglu_ffn.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/attention.py -------------------------------------------------------------------------------- /ADD/layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/block.py -------------------------------------------------------------------------------- /ADD/layers/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/dino_head.py -------------------------------------------------------------------------------- /ADD/layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/drop_path.py -------------------------------------------------------------------------------- /ADD/layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/layer_scale.py -------------------------------------------------------------------------------- /ADD/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/mlp.py -------------------------------------------------------------------------------- /ADD/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/patch_embed.py -------------------------------------------------------------------------------- /ADD/layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/layers/swiglu_ffn.py -------------------------------------------------------------------------------- /ADD/models/__pycache__/discriminator.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/models/__pycache__/discriminator.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/models/__pycache__/vit.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/models/__pycache__/vit.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/models/discriminator.py -------------------------------------------------------------------------------- /ADD/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/models/vit.py -------------------------------------------------------------------------------- /ADD/th_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/__init__.py -------------------------------------------------------------------------------- /ADD/th_utils/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/th_utils/__pycache__/custom_ops.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/__pycache__/custom_ops.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/th_utils/__pycache__/misc.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/__pycache__/misc.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/th_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/custom_ops.py -------------------------------------------------------------------------------- /ADD/th_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/misc.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/__init__.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/th_utils/ops/__pycache__/bias_act.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/__pycache__/bias_act.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/th_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /ADD/th_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/bias_act.h -------------------------------------------------------------------------------- /ADD/th_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/bias_act.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu.h -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/fma.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /ADD/th_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /ADD/th_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /ADD/th_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /ADD/th_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/th_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /ADD/utils/__pycache__/util_net.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/utils/__pycache__/util_net.cpython-310.pyc -------------------------------------------------------------------------------- /ADD/utils/util_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/ADD/utils/util_net.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/README.md -------------------------------------------------------------------------------- /dataloaders/__pycache__/paired_dataset_txt.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/dataloaders/__pycache__/paired_dataset_txt.cpython-310.pyc -------------------------------------------------------------------------------- /dataloaders/__pycache__/realesrgan.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/dataloaders/__pycache__/realesrgan.cpython-310.pyc -------------------------------------------------------------------------------- /dataloaders/paired_dataset_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/dataloaders/paired_dataset_txt.py -------------------------------------------------------------------------------- /dataloaders/params_ccsr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/dataloaders/params_ccsr.yml -------------------------------------------------------------------------------- /dataloaders/realesrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/dataloaders/realesrgan.py -------------------------------------------------------------------------------- /figs/compare_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_1.png -------------------------------------------------------------------------------- /figs/compare_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_2.png -------------------------------------------------------------------------------- /figs/compare_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_3.png -------------------------------------------------------------------------------- /figs/compare_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_4.png -------------------------------------------------------------------------------- /figs/compare_efficient.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_efficient.png -------------------------------------------------------------------------------- /figs/compare_standard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/compare_standard.png -------------------------------------------------------------------------------- /figs/fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/fig.png -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/framework.png -------------------------------------------------------------------------------- /figs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/logo.png -------------------------------------------------------------------------------- /figs/table.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/figs/table.png -------------------------------------------------------------------------------- /models/DiffAugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/DiffAugment.py -------------------------------------------------------------------------------- /models/__pycache__/DiffAugment.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/DiffAugment.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/controlnet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/controlnet.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/shared.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/shared.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/unet_2d_blocks.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/unet_2d_blocks.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/unet_2d_condition.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/unet_2d_condition.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/vit_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/__pycache__/vit_utils.cpython-310.pyc -------------------------------------------------------------------------------- /models/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/controlnet.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from models.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /models/losses/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /models/losses/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/losses/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/losses/__pycache__/contperceptual.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/contperceptual.cpython-310.pyc -------------------------------------------------------------------------------- /models/losses/__pycache__/contperceptual.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/contperceptual.cpython-37.pyc -------------------------------------------------------------------------------- /models/losses/__pycache__/contperceptual.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/__pycache__/contperceptual.cpython-38.pyc -------------------------------------------------------------------------------- /models/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/contperceptual.py -------------------------------------------------------------------------------- /models/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/losses/vqperceptual.py -------------------------------------------------------------------------------- /models/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/shared.py -------------------------------------------------------------------------------- /models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/unet_2d_condition.py -------------------------------------------------------------------------------- /models/vit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/models/vit_utils.py -------------------------------------------------------------------------------- /myutils/__pycache__/devices.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/__pycache__/devices.cpython-310.pyc -------------------------------------------------------------------------------- /myutils/__pycache__/wavelet_color_fix.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/__pycache__/wavelet_color_fix.cpython-310.pyc -------------------------------------------------------------------------------- /myutils/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/devices.py -------------------------------------------------------------------------------- /myutils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/img_util.py -------------------------------------------------------------------------------- /myutils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/misc.py -------------------------------------------------------------------------------- /myutils/vaehook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/vaehook.py -------------------------------------------------------------------------------- /myutils/wavelet_color_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/myutils/wavelet_color_fix.py -------------------------------------------------------------------------------- /pipelines/__pycache__/pipeline_ccsr.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/pipelines/__pycache__/pipeline_ccsr.cpython-310.pyc -------------------------------------------------------------------------------- /pipelines/pipeline_ccsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/pipelines/pipeline_ccsr.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/get_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/get_path.py -------------------------------------------------------------------------------- /scripts/test/test_ccsr_multistep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/test/test_ccsr_multistep.sh -------------------------------------------------------------------------------- /scripts/test/test_ccsr_onestep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/test/test_ccsr_onestep.sh -------------------------------------------------------------------------------- /scripts/test/test_ccsr_tile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/test/test_ccsr_tile.sh -------------------------------------------------------------------------------- /scripts/train/train_ccsr_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/train/train_ccsr_stage1.sh -------------------------------------------------------------------------------- /scripts/train/train_ccsr_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/train/train_ccsr_stage2.sh -------------------------------------------------------------------------------- /scripts/train/train_controlnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/scripts/train/train_controlnet.sh -------------------------------------------------------------------------------- /test_ccsr_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/test_ccsr_tile.py -------------------------------------------------------------------------------- /train_ccsr_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/train_ccsr_stage1.py -------------------------------------------------------------------------------- /train_ccsr_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/train_ccsr_stage2.py -------------------------------------------------------------------------------- /train_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/train_controlnet.py -------------------------------------------------------------------------------- /utils/__pycache__/vaehook.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/__pycache__/vaehook.cpython-310.pyc -------------------------------------------------------------------------------- /utils/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/devices.py -------------------------------------------------------------------------------- /utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/img_util.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/vaehook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/vaehook.py -------------------------------------------------------------------------------- /utils/wavelet_color_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csslc/CCSR/HEAD/utils/wavelet_color_fix.py --------------------------------------------------------------------------------