/MemoryNet

Code for paper: Memory Augment is All Your Need for image restoration(cloud,rain,shadow removal, low-light image enhancement, image deblur)即插即用提点的记忆模块

Primary LanguagePythonMIT LicenseMIT

MemoryNet:Memory augument is All You Need for image restoration

Memory augument is All You Need for image restoration Memory是一个即插即用的记忆模块,随时提点! Paper link https://arxiv.org/abs/2309.01377

The structure of MemoryNet

image

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