conda create -n cmnet
conda activate cmnet
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
pip install matplotlib scikit-image opencv-python yacs joblib natsort h5py tqdm
Install warmup scheduler
cd pytorch-gradual-warmup-lr; python setup.py install; cd ..
python train.py
python test_cloud.py
evaluate_PSNR_SSIM.m
If you use CMNet, please consider citing:
@ARTICLE{10466744, author={Liu, Jiao and Pan, Bin and Shi, Zhenwei}, journal={IEEE Transactions on Geoscience and Remote Sensing}, title={Cascaded Memory Network for Optical Remote Sensing Imagery Cloud Removal}, year={2024} }
This project is mainly based on https://github.com/swz30/MPRNet