search

SaoYan / DnCNN-PyTorch

PyTorch implementation of the TIP2017 paper "Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising"
http://ieeexplore.ieee.org/document/7839189/
GNU General Public License v3.0
401 stars 115 forks source link

FFDNet in PyTorch #2

Closed TouqeerAhmad closed 3 years ago

TouqeerAhmad commented 6 years ago

Hi SaoYan,

I was wondering if by any chance you are working or already have PyTorch version of FFDNet. It is an advanced version of DnCNN -- but only MatConvNet version is available from author's GitHub.

https://github.com/cszn/FFDNet

Thanks, Touqeer

SaoYan commented 6 years ago

Hello Touqeer,

I'm afraid I don't have time to implement it recently, but I'd be glad to do it after finishing my recent work.

fantasyzl commented 6 years ago

Hi SaoYan Thank you very much for open source related code. I have a little problem when I run the program. When I run train.py, there will always be "compare_psnr() got an unexpected keyword argument ‘data_range’”.

Thanks

RoyiAvital commented 6 years ago

I'm afraid I don't have time to implement it recently, but I'd be glad to do it after finishing my recent work.

That would be amazing. Especially if we can improve it and make it Correlated Noise aware.