KAIR icon indicating copy to clipboard operation
KAIR copied to clipboard

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

Results 63 KAIR issues
Sort by recently updated
recently updated
newest added

Thanks for @cszn contribution, DPSR is an amazing job, so I give author a star to say thanks! But in my training,It takes a long time. When I checked, I...

I want to train SwinIR using my own HR images and LR images, however it seems that it will raise error about the image size. My own HR and LR...

Hello, author. I found the following error when testing drunet under your project file. How can I solve it? `RuntimeError: Error(s) in loading state_dict for UNetRes: Missing key(s) in state_dict:...

Hello, I wonder where the "main_train_srmd.py" file is?

Where are the script and JSON file used for inference? THank U.

Could you kindly provide benchmark data related to fps and PSNR?

Would make it easier to use the code, by installing it as a library

We cannot find the training code of the deblocking model of DPIR to train deblocking model. Please provide the training code of the deblocking. Thank you very much!

As I replicate SwinIR SR x3&x4 on DIV2K dataset, I encountered a drop on U100&M109 testsets compared to results reported in the paper: All training are done with the original...

There seems to be a dependency issue of cuda kernel while using deform_attn if the current dependencies where followed. ``` RuntimeError: Error building extension 'deform_attn': [1/2] c++ -MMD -MF deform_attn_cuda_pt110.o.d...