├── .gitignore ├── README.md ├── config ├── shadow_ISTD.json ├── shadow_ISTDplus.json └── shadow_SRD.json ├── core ├── __pycache__ │ ├── logger.cpython-37.pyc │ ├── logger.cpython-39.pyc │ ├── metrics.cpython-37.pyc │ ├── metrics.cpython-39.pyc │ ├── wandb_logger.cpython-37.pyc │ └── wandb_logger.cpython-39.pyc ├── environment.txt ├── environment.yml ├── logger.py ├── metrics.py └── wandb_logger.py ├── data ├── LRHR_dataset.py ├── SRD_dataset.py ├── __init__.py ├── __pycache__ │ ├── LRHR_dataset.cpython-37.pyc │ ├── LRHR_dataset.cpython-39.pyc │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-39.pyc │ ├── util.cpython-37.pyc │ └── util.cpython-39.pyc ├── prepare_data.py └── util.py ├── doc └── framework.jpg ├── eval.py ├── infer.py ├── model ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-39.pyc │ ├── base_model.cpython-37.pyc │ ├── base_model.cpython-39.pyc │ ├── model.cpython-37.pyc │ ├── model.cpython-39.pyc │ ├── networks.cpython-37.pyc │ └── networks.cpython-39.pyc ├── base_model.py ├── ddim │ ├── ddm.py │ ├── restoration.py │ └── unet.py ├── ddpm_modules │ ├── diffusion.py │ ├── unet.py │ └── vanilla_unet.py ├── model.py ├── networks.py └── sr3_modules │ ├── __pycache__ │ ├── diffusion.cpython-37.pyc │ ├── diffusion.cpython-39.pyc │ ├── transformer.cpython-37.pyc │ ├── unet.cpython-37.pyc │ └── unet.cpython-39.pyc │ ├── diffusion.py │ ├── skip_unet.py │ ├── transformer.py │ ├── unet.py │ └── wavagrad.py ├── requirements.txt ├── sample.py ├── sr.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-37.pyc ├── dataset_utils.cpython-37.pyc ├── dir_utils.cpython-37.pyc ├── image_utils.cpython-37.pyc └── model_utils.cpython-37.pyc ├── antialias.py ├── bundle_submissions.py ├── dataset_utils.py ├── dir_utils.py ├── illum_affine_model.py ├── image_utils.py ├── loader.py ├── model_utils.py ├── optimize.py └── util.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.log 2 | *.pth 3 | 4 | experiments/ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/README.md -------------------------------------------------------------------------------- /config/shadow_ISTD.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/config/shadow_ISTD.json -------------------------------------------------------------------------------- /config/shadow_ISTDplus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/config/shadow_ISTDplus.json -------------------------------------------------------------------------------- /config/shadow_SRD.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/config/shadow_SRD.json -------------------------------------------------------------------------------- /core/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/logger.cpython-39.pyc -------------------------------------------------------------------------------- /core/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/metrics.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/metrics.cpython-39.pyc -------------------------------------------------------------------------------- /core/__pycache__/wandb_logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/wandb_logger.cpython-37.pyc -------------------------------------------------------------------------------- /core/__pycache__/wandb_logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/__pycache__/wandb_logger.cpython-39.pyc -------------------------------------------------------------------------------- /core/environment.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/environment.txt -------------------------------------------------------------------------------- /core/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/environment.yml -------------------------------------------------------------------------------- /core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/logger.py -------------------------------------------------------------------------------- /core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/metrics.py -------------------------------------------------------------------------------- /core/wandb_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/core/wandb_logger.py -------------------------------------------------------------------------------- /data/LRHR_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/LRHR_dataset.py -------------------------------------------------------------------------------- /data/SRD_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/SRD_dataset.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/__pycache__/LRHR_dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/LRHR_dataset.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/LRHR_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/LRHR_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /data/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /data/__pycache__/util.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/util.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/__pycache__/util.cpython-39.pyc -------------------------------------------------------------------------------- /data/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/prepare_data.py -------------------------------------------------------------------------------- /data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/data/util.py -------------------------------------------------------------------------------- /doc/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/doc/framework.jpg -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/eval.py -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/infer.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/base_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/base_model.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/base_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/base_model.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/model.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/networks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/networks.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/networks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/__pycache__/networks.cpython-39.pyc -------------------------------------------------------------------------------- /model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/base_model.py -------------------------------------------------------------------------------- /model/ddim/ddm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddim/ddm.py -------------------------------------------------------------------------------- /model/ddim/restoration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddim/restoration.py -------------------------------------------------------------------------------- /model/ddim/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddim/unet.py -------------------------------------------------------------------------------- /model/ddpm_modules/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddpm_modules/diffusion.py -------------------------------------------------------------------------------- /model/ddpm_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddpm_modules/unet.py -------------------------------------------------------------------------------- /model/ddpm_modules/vanilla_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/ddpm_modules/vanilla_unet.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/model.py -------------------------------------------------------------------------------- /model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/networks.py -------------------------------------------------------------------------------- /model/sr3_modules/__pycache__/diffusion.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/__pycache__/diffusion.cpython-37.pyc -------------------------------------------------------------------------------- /model/sr3_modules/__pycache__/diffusion.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/__pycache__/diffusion.cpython-39.pyc -------------------------------------------------------------------------------- /model/sr3_modules/__pycache__/transformer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/__pycache__/transformer.cpython-37.pyc -------------------------------------------------------------------------------- /model/sr3_modules/__pycache__/unet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/__pycache__/unet.cpython-37.pyc -------------------------------------------------------------------------------- /model/sr3_modules/__pycache__/unet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/__pycache__/unet.cpython-39.pyc -------------------------------------------------------------------------------- /model/sr3_modules/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/diffusion.py -------------------------------------------------------------------------------- /model/sr3_modules/skip_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/skip_unet.py -------------------------------------------------------------------------------- /model/sr3_modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/transformer.py -------------------------------------------------------------------------------- /model/sr3_modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/unet.py -------------------------------------------------------------------------------- /model/sr3_modules/wavagrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/model/sr3_modules/wavagrad.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/sample.py -------------------------------------------------------------------------------- /sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/sr.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dataset_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__pycache__/dataset_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dir_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__pycache__/dir_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/image_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__pycache__/image_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/model_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/__pycache__/model_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/antialias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/antialias.py -------------------------------------------------------------------------------- /utils/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/bundle_submissions.py -------------------------------------------------------------------------------- /utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/dataset_utils.py -------------------------------------------------------------------------------- /utils/dir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/dir_utils.py -------------------------------------------------------------------------------- /utils/illum_affine_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/illum_affine_model.py -------------------------------------------------------------------------------- /utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/image_utils.py -------------------------------------------------------------------------------- /utils/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/loader.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/optimize.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GuoLanqing/ShadowDiffusion/HEAD/utils/util.py --------------------------------------------------------------------------------