Awesome
Diffusion-Enhancement-for-CR
🔥 Congradualation to my collaborators and myself! This paper has been accepted by IEEE Transactions on Geoscience and Remote Sensing! It's a new start!
The code for Diffusion Enhancement for Cloud Removal in Ultra-Resolution Remote Sensing Imagery, which is based on ADM.
Several conventional CR models could refer to https://github.com/littlebeen/Cloud-removal-model-collection!
Usage
- Data preparation
Download the dataset and put it into ./data
- pretrain CR model (The reference visual prior mentioned in paper)
Download the pretain e2e model (mdsa/mn as your need) and put it into guided_diffusion/couldnet/mdsa/pretrain or guided_diffusion/couldnet/mn/pretrain respectively (mdsaxx put in mdsa/pretrain, mnxx put in mn/pretrain). The model could be found at https://pan.baidu.com/s/1lyKRG67AxM5SZEj3mhTKWA code:bean
You could choose the appropriate pretrain model as your need (Which model You want to train or test based on? memorynet(mn) or mdsa?). Must be done before training and testing!
Train
- Pure diffusion.
change guided_diffusion/respace.py line4/5 to gaussian_diffusion
change guided_diffusion/unet.py line1140 to 'class SuperResModel(UNetModel):'
- Locked diffusion + Trained WA
change guided_diffusion/respace.py line4/5 to gaussian_diffusiongaussian_diffusion_enhance
change guided_diffusion/unet.py 'class SuperResModel(UNetModel256):'
locked in guided_diffusion/train_util.py line74
- ALL-in change train_util.py line74
python super_res_train.py
Test
- Put the pre-train model into 'pre_train'
python super_res_sample.py
Weight
Our pre-train models on RICE2 with mn and mdsa are uploaded. https://pan.baidu.com/s/1SvnPL7HRKqSK0zDixYpHow code:bean
CUHK-CR
A novel multispectral Cloud Removal dataset
Download link: https://pan.baidu.com/s/1z2SgORYz5_t94kya8CeqiQ code:bean
-CUHK-CR1 (The RGB images for thin dataset CUHK-CR1)
-CUHK-CR2 (The RGB images for think dataset CUHK-CR2)
-nir (the near-infrared images for CUHK-CR1 and CUHK-CR2)
If you need image with 4 bands (RGB + nir), you could load the images in the RGB dataset and the nir dataset and combine the 4 channels together.
Cite
If this project is useful to you, please cite this paper :)
@article{sui2024diffusion,
author={Sui, Jialu and Ma, Yiyang and Yang, Wenhan and Zhang, Xiaokang and Pun, Man-On and Liu, Jiaying},
journal={IEEE Transactions on Geoscience and Remote Sensing},
title={Diffusion Enhancement for Cloud Removal in Ultra-Resolution Remote Sensing Imagery},
year={2024},
volume={62},
pages={1-14},
doi={10.1109/TGRS.2024.3411671}}
If you have any question, be free to contact with me!