├── .DS_Store ├── .idea ├── .gitignore ├── Learning-Hazing-to-Dehazing.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.md ├── assets ├── .DS_Store ├── flowchart.jpg ├── result.jpg └── teaser.jpg ├── configs ├── .DS_Store ├── inference │ ├── .DS_Store │ ├── stage1.yaml │ ├── stage2.yaml │ └── stage2_accsamp.yaml └── train │ ├── .DS_Store │ ├── stage1.yaml │ └── stage2.yaml ├── datasets.py ├── diffbir ├── .DS_Store ├── model │ ├── .DS_Store │ ├── __init__.py │ ├── attention.py │ ├── cldm.py │ ├── clip.py │ ├── config.py │ ├── controlnet.py │ ├── distributions.py │ ├── gaussian_diffusion.py │ ├── open_clip │ │ ├── __init__.py │ │ ├── bpe_simple_vocab_16e6.txt.gz │ │ ├── model.py │ │ ├── tokenizer.py │ │ └── transformer.py │ ├── unet.py │ ├── util.py │ └── vae.py ├── pipeline.py ├── sampler │ ├── .DS_Store │ ├── __init__.py │ ├── sampler.py │ └── spaced_sampler.py └── utils │ ├── .DS_Store │ ├── common.py │ ├── cond_fn.py │ └── tilevae │ ├── __init__.py │ ├── attn.py │ └── tilevae.py ├── inference_accsamp.py ├── inference_stage1.py ├── inference_stage2.py ├── inputs ├── .DS_Store ├── 1.png ├── 2.png ├── 4.png ├── 5.png ├── 6.png └── 7.png ├── requirements.txt ├── train_stage1.py ├── train_stage2.py └── utils ├── __init__.py ├── align_utils.py ├── data_utils.py ├── dcp_utils.py └── ssim_utils.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.DS_Store -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/Learning-Hazing-to-Dehazing.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/Learning-Hazing-to-Dehazing.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/README.md -------------------------------------------------------------------------------- /assets/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/assets/.DS_Store -------------------------------------------------------------------------------- /assets/flowchart.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/assets/flowchart.jpg -------------------------------------------------------------------------------- /assets/result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/assets/result.jpg -------------------------------------------------------------------------------- /assets/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/assets/teaser.jpg -------------------------------------------------------------------------------- /configs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/.DS_Store -------------------------------------------------------------------------------- /configs/inference/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/inference/.DS_Store -------------------------------------------------------------------------------- /configs/inference/stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/inference/stage1.yaml -------------------------------------------------------------------------------- /configs/inference/stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/inference/stage2.yaml -------------------------------------------------------------------------------- /configs/inference/stage2_accsamp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/inference/stage2_accsamp.yaml -------------------------------------------------------------------------------- /configs/train/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/train/.DS_Store -------------------------------------------------------------------------------- /configs/train/stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/train/stage1.yaml -------------------------------------------------------------------------------- /configs/train/stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/configs/train/stage2.yaml -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/datasets.py -------------------------------------------------------------------------------- /diffbir/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/.DS_Store -------------------------------------------------------------------------------- /diffbir/model/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/.DS_Store -------------------------------------------------------------------------------- /diffbir/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/__init__.py -------------------------------------------------------------------------------- /diffbir/model/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/attention.py -------------------------------------------------------------------------------- /diffbir/model/cldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/cldm.py -------------------------------------------------------------------------------- /diffbir/model/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/clip.py -------------------------------------------------------------------------------- /diffbir/model/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/config.py -------------------------------------------------------------------------------- /diffbir/model/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/controlnet.py -------------------------------------------------------------------------------- /diffbir/model/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/distributions.py -------------------------------------------------------------------------------- /diffbir/model/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/gaussian_diffusion.py -------------------------------------------------------------------------------- /diffbir/model/open_clip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/open_clip/__init__.py -------------------------------------------------------------------------------- /diffbir/model/open_clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/open_clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /diffbir/model/open_clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/open_clip/model.py -------------------------------------------------------------------------------- /diffbir/model/open_clip/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/open_clip/tokenizer.py -------------------------------------------------------------------------------- /diffbir/model/open_clip/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/open_clip/transformer.py -------------------------------------------------------------------------------- /diffbir/model/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/unet.py -------------------------------------------------------------------------------- /diffbir/model/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/util.py -------------------------------------------------------------------------------- /diffbir/model/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/model/vae.py -------------------------------------------------------------------------------- /diffbir/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/pipeline.py -------------------------------------------------------------------------------- /diffbir/sampler/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/sampler/.DS_Store -------------------------------------------------------------------------------- /diffbir/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/sampler/__init__.py -------------------------------------------------------------------------------- /diffbir/sampler/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/sampler/sampler.py -------------------------------------------------------------------------------- /diffbir/sampler/spaced_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/sampler/spaced_sampler.py -------------------------------------------------------------------------------- /diffbir/utils/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/utils/.DS_Store -------------------------------------------------------------------------------- /diffbir/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/utils/common.py -------------------------------------------------------------------------------- /diffbir/utils/cond_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/utils/cond_fn.py -------------------------------------------------------------------------------- /diffbir/utils/tilevae/__init__.py: -------------------------------------------------------------------------------- 1 | from .tilevae import VAEHook 2 | -------------------------------------------------------------------------------- /diffbir/utils/tilevae/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/utils/tilevae/attn.py -------------------------------------------------------------------------------- /diffbir/utils/tilevae/tilevae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/diffbir/utils/tilevae/tilevae.py -------------------------------------------------------------------------------- /inference_accsamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inference_accsamp.py -------------------------------------------------------------------------------- /inference_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inference_stage1.py -------------------------------------------------------------------------------- /inference_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inference_stage2.py -------------------------------------------------------------------------------- /inputs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/.DS_Store -------------------------------------------------------------------------------- /inputs/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/1.png -------------------------------------------------------------------------------- /inputs/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/2.png -------------------------------------------------------------------------------- /inputs/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/4.png -------------------------------------------------------------------------------- /inputs/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/5.png -------------------------------------------------------------------------------- /inputs/6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/6.png -------------------------------------------------------------------------------- /inputs/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/inputs/7.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/train_stage1.py -------------------------------------------------------------------------------- /train_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/train_stage2.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/align_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/utils/align_utils.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/dcp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/utils/dcp_utils.py -------------------------------------------------------------------------------- /utils/ssim_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ruiyi-w/Learning-Hazing-to-Dehazing/HEAD/utils/ssim_utils.py --------------------------------------------------------------------------------