cszn / KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

Home Page:https://cszn.github.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training FFDNet

shengyenlin opened this issue · comments

Hi,

Thanks for you amazing work

I would like to ask how could I train the FFDNet on my customized dataset.

I saw we could download the pretrained model and test it, but where's the training code for it?

I know there's matlab version, but is there pytorch version?

Thanks in advance.