KAIR
KAIR copied to clipboard
Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR
Great paper and great code! I didn't find codes correspond to the feature “spatially variant degradation”, and I am also confused about how "Dimensionality Stretching" strategy can involve spatial information......
Hi @cszn, I've trained DPSR for my own dataset and the results seem pretty good when training. If I apply the learned weights to some new images (very similar to...
There isn't any training code available for the blind DnCNN in either this repository or the DnCNN one. I was wondering if this could be provided.
Hi Thanks for the great work again. I am wondering whether there might be something wrong with the learning rate scheduler. I tested "main_train_rrdb_psnr.py", found that the learning rate quickly...
Hey! Thanks for your contribution! This could help me a lot for my jobs ! But now I have an issue so I hope you can answer me for my...
Getting this issue while running the training code. I guess it's because of the missing h_nc value in the json. What h_nc value should be kept for the model? ![Screenshot...
Hi, Thanks for this repository. I am trying to retrain SRMD x4 on DIV2K so I changed options/train_srmd.json with my database path and changed the frequency of the checkpoints. After...
Hi! Thank you so much for making this work open source! I have a custom dataset (train/test) which I would like to use for training and testing. I was able...
Hi First of all, i'm very appreciate to your nice work. When I tried to train the ffdnet it stucked in dataloader in epoch "for loop". However, it works in...