├── README.md ├── configs ├── raindrop.yml └── raindrop_wavelet.yml ├── datasets ├── .DS_Store ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── dataset.cpython-38.pyc │ └── raindrop.cpython-38.pyc ├── dataset.py └── raindrop.py ├── eval_diffusion.py ├── models ├── .DS_Store ├── Lap.py ├── __init__.py ├── __pycache__ │ ├── NAFNet.cpython-38.pyc │ ├── __init__.cpython-38.pyc │ ├── arch.cpython-38.pyc │ ├── ddm.cpython-38.pyc │ ├── ddm_wavelet.cpython-38.pyc │ ├── model_dense.cpython-38.pyc │ ├── restoration.cpython-38.pyc │ ├── sam.cpython-38.pyc │ ├── unet.cpython-38.pyc │ └── wavelet.cpython-38.pyc ├── arch.py ├── ddm.py ├── ddm_wavelet.py ├── model_dense.py ├── restoration.py ├── sam.py ├── unet.py ├── unet_wav.py ├── wavelet.py └── wavelet_weights_c2.pkl ├── requirements.txt ├── results └── .DS_Store ├── test_weather_script.py ├── train_diffusion.py ├── train_hfrm.py ├── train_result └── .DS_Store ├── train_weather_script.py └── utils ├── .DS_Store ├── __init__.py ├── __pycache__ ├── __init__.cpython-38.pyc ├── logging.cpython-38.pyc ├── metrics.cpython-38.pyc ├── optimize.cpython-38.pyc └── sampling.cpython-38.pyc ├── logging.py ├── metrics.py ├── optimize.py └── sampling.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/README.md -------------------------------------------------------------------------------- /configs/raindrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/configs/raindrop.yml -------------------------------------------------------------------------------- /configs/raindrop_wavelet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/configs/raindrop_wavelet.yml -------------------------------------------------------------------------------- /datasets/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/.DS_Store -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/__pycache__/dataset.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/raindrop.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/__pycache__/raindrop.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/dataset.py -------------------------------------------------------------------------------- /datasets/raindrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/datasets/raindrop.py -------------------------------------------------------------------------------- /eval_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/eval_diffusion.py -------------------------------------------------------------------------------- /models/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/.DS_Store -------------------------------------------------------------------------------- /models/Lap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/Lap.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/__pycache__/NAFNet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/NAFNet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/arch.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/ddm.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/ddm.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/ddm_wavelet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/ddm_wavelet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/model_dense.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/model_dense.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/restoration.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/restoration.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/sam.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/unet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/unet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/wavelet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/__pycache__/wavelet.cpython-38.pyc -------------------------------------------------------------------------------- /models/arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/arch.py -------------------------------------------------------------------------------- /models/ddm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/ddm.py -------------------------------------------------------------------------------- /models/ddm_wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/ddm_wavelet.py -------------------------------------------------------------------------------- /models/model_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/model_dense.py -------------------------------------------------------------------------------- /models/restoration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/restoration.py -------------------------------------------------------------------------------- /models/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/sam.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/unet.py -------------------------------------------------------------------------------- /models/unet_wav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/unet_wav.py -------------------------------------------------------------------------------- /models/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/wavelet.py -------------------------------------------------------------------------------- /models/wavelet_weights_c2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/models/wavelet_weights_c2.pkl -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/results/.DS_Store -------------------------------------------------------------------------------- /test_weather_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/test_weather_script.py -------------------------------------------------------------------------------- /train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/train_diffusion.py -------------------------------------------------------------------------------- /train_hfrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/train_hfrm.py -------------------------------------------------------------------------------- /train_result/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/train_result/.DS_Store -------------------------------------------------------------------------------- /train_weather_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/train_weather_script.py -------------------------------------------------------------------------------- /utils/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/.DS_Store -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logging.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__pycache__/logging.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/optimize.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__pycache__/optimize.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/sampling.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/__pycache__/sampling.cpython-38.pyc -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/optimize.py -------------------------------------------------------------------------------- /utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Easquel/WaveDM/HEAD/utils/sampling.py --------------------------------------------------------------------------------