KAIR
KAIR copied to clipboard
Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR
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...