cszn / FFDNet

FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoising (TIP, 2018)

Home Page:https://ieeexplore.ieee.org/abstract/document/8365806/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch version is very slow compared with matconvnet

kunkun007 opened this issue · comments

I use your matlab version and pytorch for test ffdnet color model, matlab version's time is about 20ms, but pytorch version's time is about 1s. Cuda version is 8.0 and cudnn is 6.0.21. Can you have some ideas about this? What is your testing time and pytorch version?