holywu / vs-swinir

SwinIR function for VapourSynth

SwinIR

Image Restoration Using Swin Transformer, based on https://github.com/JingyunLiang/SwinIR.

Dependencies

• NumPy
• PyTorch 1.13
• VapourSynth R55+

Installation

pip install -U vsswinir
python -m vsswinir

Usage

from vsswinir import swinir

ret = swinir(clip)

See __init__.py for the description of the parameters.

vs-swinir's Issues

Support for SwinIR-M_x2_GAN.pth?

Hi, @HolyWu
Thanks for great work.
Could you please implement
003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth
model.?
Thank you very much!

Omni-SR

Can https://github.com/Francis0625/Omni-SR
Is it imported into vs? It looks better than swinir.

denoising support

how about add denoising support? the result looks better than dpir in my test:smile:

unsupported operand type(s)

I used 2.0:
script:

ret= core.lsmas.LWLibavSource(r'D:\video\VapourSynth64Portable\1.mp4')
core.resize.Bicubic(ret, format=vs.RGBS,dither_type="error_diffusion")


ret =SwinIR(ret,model=0)


ret = core.resize.Bicubic(ret, format=vs.YUV420P16, matrix_s="709", range_s="full")



ret = core.resize.Bicubic(ret, format=vs.YUV420P8,matrix_s="709",range_s="full")




ret.set_output()>

result:

Failed to evaluate the script:
Python exception: unsupported operand type(s) for //: 'vapoursynth.VideoNode' and 'int'

Traceback (most recent call last):
File "src\cython\vapoursynth.pyx", line 2866, in vapoursynth._vpy_evaluate
File "src\cython\vapoursynth.pyx", line 2867, in vapoursynth._vpy_evaluate
File "D:/video/VapourSynth64Portablevb/swinir2.vpy", line 46, in 
ret =SwinIR(ret,model=0)
File "D:\video\VapourSynth64Portablevb\VapourSynth64\Lib\site-packages\vsswinir\network_swinir.py", line 561, in __init__
self.patch_embed = PatchEmbed(
File "D:\video\VapourSynth64Portablevb\VapourSynth64\Lib\site-packages\vsswinir\network_swinir.py", line 406, in __init__
patches_resolution = [img_size[0] // patch_size[0], img_size[1] // patch_size[1]]
TypeError: unsupported operand type(s) for //: 'vapoursynth.VideoNode' and 'int'>

but 1.0 is ok!