Skip to content
forked from tyshiwo/MemNet

Code for our ICCV'17 paper "MemNet: A Persistent Memory Network for Image Restoration" (SPOTLIGHT Presentation)

Notifications You must be signed in to change notification settings

ironfist2/MemNet

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MemNet

Citation

If you find MemNet useful in your research, please consider citing:

@inproceedings{Tai-MemNet-2017,
  title={MemNet: A Persistent Memory Network for Image Restoration},
  author={Tai, Ying and Yang, Jian and Liu, Xiaoming and Xu, Chunyan},
  booktitle={Proceedings of International Conference on Computer Vision},
  year={2017}
}

Implement adjustable gradient clipping

modify sgd_solver.cpp in your_caffe_root/src/caffe/solvers/, where we add the following codes in funciton ClipGradients():

Dtype rate = GetLearningRate();

const Dtype clip_gradients = this->param_.clip_gradients()/rate;

Training (Taking Super-resolution task as the example)

  1. Preparing training/validation data using the files: generate_trainingset_x234/generate_testingset_x234 in "data/SuperResolution" folder. "Train_291" folder contains 291 training images and "Set5" folder is a popular benchmark dataset.

  2. We release two MemNet architectures: MemNet_M6R6_80C64 and MemNet_M10R10_212C64 in "caffe_files" folder. Choose either one to do training.

     $ cd ./caffe_files/MemNet_M6R6_80C64
     $ ./train_MemNet_M6R6_80C64.sh
    

Test (Taking Super-resolution task as the example)

  1. Remember to compile the matlab wrapper: make matcaffe, since we use matlab to do testing.

  2. We release two pretrained models: MemNet_M6R6_80C64 and MemNet_M10R10_212C64 in "model" folder. Choose either one to do testing on benchmark Set5.

     $ cd ./results/MemNet_M6R6_80C64
     $ matlab
     >> test_MemNet_M6R6_SR
    

    The results are stored in "results" folder, with both reconstructed images and PSNR/SSIMs.

More Qualitative results

Image denoising

Super-resolution

JPEG deblocking

About

Code for our ICCV'17 paper "MemNet: A Persistent Memory Network for Image Restoration" (SPOTLIGHT Presentation)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MATLAB 56.0%
  • HTML 24.3%
  • C 16.7%
  • C++ 2.7%
  • CSS 0.2%
  • Shell 0.1%