cszn / DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)
https://cszn.github.io/
1.47k stars 539 forks source link

About the usage of DnCNN with matconvnet-1.0-beta24 on linux #5

Closed ZhangJingrong closed 7 years ago

ZhangJingrong commented 7 years ago

Thanks so much for sharing your code. I tried to use it and find two problems:

  1. The slash in the path on Linux is different from Windows:
     folderTest  = 'testsets\Set12'; %%% test dataset
  2. For "matconvnet" version later than 20, some modifications need to be made. A bug " missing field 'dilate' in layers" shows up. I tried to fix this by using
    net = vl_simplenn_tidy(net)
    net = vl_simplenn_move(net,'gpu');

    before

    res    = vl_simplenn(net,input,[],[],'conserveMemory',true,'mode','test');

    Now it works.

cszn commented 7 years ago

Thanks. ^_^