/CB-MemoryNet

Code for paper: CB-MemoryNet:Contrastive balance learning memory net for image restoration(cloud,rain,shadow removal))

Primary LanguagePythonMIT LicenseMIT

CB-MemoryNet:Contrastive balance learning memory net for image restoration

CB-MemoryNet:Contrastive balance learning memory net for image restoration，image shadow removal, rain removal, debulrring.

Results of Shadow removal on ISTD dataset

image

Quick Run

To test the pre-trained models of Decloud on your own images, run

python demo.py --task Task_Name --input_dir path_to_images --result_dir save_images_here

Pretrained model

  1. Download the pretrained model cloud-removal

2.Baidu Drive: 链接：https://pan.baidu.com/s/1nBNEsRLIFS2VVtHl8O14Rw 提取码：5mli

Dataset

Download datasets RICE from here, and ISTD dataset from here

To reproduce PSNR/SSIM scores of the paper, run MATLAB script

evaluate_PSNR_SSIM.m

ACKNOLAGEMENT

The code is updated on https://github.com/swz30/MPRNet