├── README.md ├── config ├── sp │ ├── SR3_sp_cosine_128.json │ ├── SR3_sp_cosine_128_finetune.json │ └── SR3_sp_cosine_128_finetune_infer.json ├── t2m │ ├── SR3_t2m_cosine_128.json │ ├── SR3_t2m_cosine_128_finetune.json │ └── SR3_t2m_cosine_128_finetune_infer.json ├── tp │ ├── SR3_tp_cosine_128.json │ ├── SR3_tp_cosine_128_finetune.json │ └── SR3_tp_cosine_128_finetune_infer.json ├── u │ ├── SR3_u.json │ ├── SR3_u_finetune.json │ └── SR3_u_finetune_infer.json └── v │ ├── SR3_v.json │ ├── SR3_v_finetune.json │ └── SR3_v_infer.json ├── configs.py ├── data ├── .ipynb_checkpoints │ └── mydataset_patch-checkpoint.py ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ └── mydataset_patch.cpython-39.pyc ├── mydataset_patch.py └── test.png ├── finetune_all.py ├── inference_2_monthly.py ├── model ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── base_model.cpython-39.pyc │ ├── ema.cpython-39.pyc │ ├── model.cpython-39.pyc │ └── networks.cpython-39.pyc ├── base_model.py ├── diffusion │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── ddim.cpython-39.pyc │ │ ├── ddpm.cpython-39.pyc │ │ └── unet.cpython-39.pyc │ ├── ddim.py │ ├── ddpm.py │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── dpm_solver.cpython-39.pyc │ │ │ └── sampler.cpython-39.pyc │ │ ├── dpm_solver.py │ │ └── sampler.py │ └── unet.py ├── ema.py ├── model.py └── networks.py ├── pytorch_ssim.py ├── src.png ├── trainer_all.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/README.md -------------------------------------------------------------------------------- /config/sp/SR3_sp_cosine_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/sp/SR3_sp_cosine_128.json -------------------------------------------------------------------------------- /config/sp/SR3_sp_cosine_128_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/sp/SR3_sp_cosine_128_finetune.json -------------------------------------------------------------------------------- /config/sp/SR3_sp_cosine_128_finetune_infer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/sp/SR3_sp_cosine_128_finetune_infer.json -------------------------------------------------------------------------------- /config/t2m/SR3_t2m_cosine_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/t2m/SR3_t2m_cosine_128.json -------------------------------------------------------------------------------- /config/t2m/SR3_t2m_cosine_128_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/t2m/SR3_t2m_cosine_128_finetune.json -------------------------------------------------------------------------------- /config/t2m/SR3_t2m_cosine_128_finetune_infer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/t2m/SR3_t2m_cosine_128_finetune_infer.json -------------------------------------------------------------------------------- /config/tp/SR3_tp_cosine_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/tp/SR3_tp_cosine_128.json -------------------------------------------------------------------------------- /config/tp/SR3_tp_cosine_128_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/tp/SR3_tp_cosine_128_finetune.json -------------------------------------------------------------------------------- /config/tp/SR3_tp_cosine_128_finetune_infer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/tp/SR3_tp_cosine_128_finetune_infer.json -------------------------------------------------------------------------------- /config/u/SR3_u.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/u/SR3_u.json -------------------------------------------------------------------------------- /config/u/SR3_u_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/u/SR3_u_finetune.json -------------------------------------------------------------------------------- /config/u/SR3_u_finetune_infer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/u/SR3_u_finetune_infer.json -------------------------------------------------------------------------------- /config/v/SR3_v.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/v/SR3_v.json -------------------------------------------------------------------------------- /config/v/SR3_v_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/v/SR3_v_finetune.json -------------------------------------------------------------------------------- /config/v/SR3_v_infer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/config/v/SR3_v_infer.json -------------------------------------------------------------------------------- /configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/configs.py -------------------------------------------------------------------------------- /data/.ipynb_checkpoints/mydataset_patch-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/data/.ipynb_checkpoints/mydataset_patch-checkpoint.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/data/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /data/__pycache__/mydataset_patch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/data/__pycache__/mydataset_patch.cpython-39.pyc -------------------------------------------------------------------------------- /data/mydataset_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/data/mydataset_patch.py -------------------------------------------------------------------------------- /data/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/data/test.png -------------------------------------------------------------------------------- /finetune_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/finetune_all.py -------------------------------------------------------------------------------- /inference_2_monthly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/inference_2_monthly.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/base_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__pycache__/base_model.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/ema.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__pycache__/ema.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__pycache__/model.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/networks.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/__pycache__/networks.cpython-39.pyc -------------------------------------------------------------------------------- /model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/base_model.py -------------------------------------------------------------------------------- /model/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/diffusion/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/__pycache__/ddim.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/__pycache__/ddim.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/__pycache__/ddpm.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/__pycache__/ddpm.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/__pycache__/unet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/__pycache__/unet.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/ddim.py -------------------------------------------------------------------------------- /model/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/ddpm.py -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/__pycache__/dpm_solver.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/__pycache__/dpm_solver.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/__pycache__/sampler.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/__pycache__/sampler.cpython-39.pyc -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /model/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /model/diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/diffusion/unet.py -------------------------------------------------------------------------------- /model/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/ema.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/model.py -------------------------------------------------------------------------------- /model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/model/networks.py -------------------------------------------------------------------------------- /pytorch_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/pytorch_ssim.py -------------------------------------------------------------------------------- /src.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/src.png -------------------------------------------------------------------------------- /trainer_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/trainer_all.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LingFH/Diffusion_4_downscaling/HEAD/utils.py --------------------------------------------------------------------------------