├── LICENSE ├── README.md ├── basicsr ├── __init__.py └── utils │ ├── __init__.py │ ├── dist_util.py │ ├── download_util.py │ └── misc.py ├── configs ├── DifIISR_test.yaml └── realsr_swinunet_realesrgan256.yaml ├── datapipe ├── __init__.py └── datasets.py ├── dataset └── test │ ├── HR │ ├── 00950.png │ ├── 01670.png │ ├── 01673.png │ ├── 01828.png │ ├── 01963.png │ ├── 01977.png │ ├── 02033.png │ ├── 02090.png │ ├── 02119.png │ └── 03836.png │ └── LR │ ├── 00950x4.png │ ├── 01670x4.png │ ├── 01673x4.png │ ├── 01828x4.png │ ├── 01963x4.png │ ├── 01977x4.png │ ├── 02033x4.png │ ├── 02090x4.png │ ├── 02119x4.png │ └── 03836x4.png ├── evaluate.py ├── figures ├── Overview.png └── network.png ├── inference.py ├── ldm ├── models │ └── autoencoder.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── model_back.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ ├── quantize.py │ └── vqvae │ │ └── quantize.py └── util.py ├── models ├── basic_ops.py ├── fp16_util.py ├── gaussian_diffusion_test.py ├── losses.py ├── resample.py ├── respace.py ├── script_util.py ├── swin_transformer.py └── unet.py ├── requirements.txt ├── sampler.py └── utils ├── __init__.py ├── util_common.py ├── util_image.py ├── util_net.py └── util_sisr.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/README.md -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- 1 | # https://github.com/xinntao/BasicSR 2 | # flake8: noqa 3 | from .utils import * 4 | -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /configs/DifIISR_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/configs/DifIISR_test.yaml -------------------------------------------------------------------------------- /configs/realsr_swinunet_realesrgan256.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/configs/realsr_swinunet_realesrgan256.yaml -------------------------------------------------------------------------------- /datapipe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/datapipe/__init__.py -------------------------------------------------------------------------------- /datapipe/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/datapipe/datasets.py -------------------------------------------------------------------------------- /dataset/test/HR/00950.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/00950.png -------------------------------------------------------------------------------- /dataset/test/HR/01670.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/01670.png -------------------------------------------------------------------------------- /dataset/test/HR/01673.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/01673.png -------------------------------------------------------------------------------- /dataset/test/HR/01828.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/01828.png -------------------------------------------------------------------------------- /dataset/test/HR/01963.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/01963.png -------------------------------------------------------------------------------- /dataset/test/HR/01977.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/01977.png -------------------------------------------------------------------------------- /dataset/test/HR/02033.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/02033.png -------------------------------------------------------------------------------- /dataset/test/HR/02090.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/02090.png -------------------------------------------------------------------------------- /dataset/test/HR/02119.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/02119.png -------------------------------------------------------------------------------- /dataset/test/HR/03836.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/HR/03836.png -------------------------------------------------------------------------------- /dataset/test/LR/00950x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/00950x4.png -------------------------------------------------------------------------------- /dataset/test/LR/01670x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/01670x4.png -------------------------------------------------------------------------------- /dataset/test/LR/01673x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/01673x4.png -------------------------------------------------------------------------------- /dataset/test/LR/01828x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/01828x4.png -------------------------------------------------------------------------------- /dataset/test/LR/01963x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/01963x4.png -------------------------------------------------------------------------------- /dataset/test/LR/01977x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/01977x4.png -------------------------------------------------------------------------------- /dataset/test/LR/02033x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/02033x4.png -------------------------------------------------------------------------------- /dataset/test/LR/02090x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/02090x4.png -------------------------------------------------------------------------------- /dataset/test/LR/02119x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/02119x4.png -------------------------------------------------------------------------------- /dataset/test/LR/03836x4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/dataset/test/LR/03836x4.png -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/evaluate.py -------------------------------------------------------------------------------- /figures/Overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/figures/Overview.png -------------------------------------------------------------------------------- /figures/network.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/figures/network.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/inference.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model_back.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/diffusionmodules/model_back.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/quantize.py -------------------------------------------------------------------------------- /ldm/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/ldm/util.py -------------------------------------------------------------------------------- /models/basic_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/basic_ops.py -------------------------------------------------------------------------------- /models/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/fp16_util.py -------------------------------------------------------------------------------- /models/gaussian_diffusion_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/gaussian_diffusion_test.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/resample.py -------------------------------------------------------------------------------- /models/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/respace.py -------------------------------------------------------------------------------- /models/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/script_util.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/models/unet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/requirements.txt -------------------------------------------------------------------------------- /sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/sampler.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/util_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/utils/util_common.py -------------------------------------------------------------------------------- /utils/util_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/utils/util_image.py -------------------------------------------------------------------------------- /utils/util_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/utils/util_net.py -------------------------------------------------------------------------------- /utils/util_sisr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zirui0625/DifIISR/HEAD/utils/util_sisr.py --------------------------------------------------------------------------------