Guest User

RealCUGAN + BasicVSR++

a guest
Apr 6th, 2026
105
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 5.79 KB | None | 0 0
  1. # Imports
  2. import ctypes
  3. import os
  4. import sys
  5. # Import site-packages
  6. sys.path.append("F:/Hybrid/64bit/Vapoursynth/Lib/site-packages")
  7. import vapoursynth as vs
  8. # getting Vapoursynth core
  9. core = vs.core
  10. # Limit frame cache to 48449MB
  11. core.max_cache_size = 48449
  12. # Import scripts folder
  13. scriptPath = 'F:/Hybrid/64bit/vsscripts'
  14. sys.path.insert(0, os.path.abspath(scriptPath))
  15. # Loading Support Files
  16. Dllref = ctypes.windll.LoadLibrary("F:/Hybrid/64bit/vsfilters/Support/libfftw3f-3.dll")
  17. # loading plugins
  18. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/ResizeFilter/RealCUGAN/realcugannv.dll")
  19. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DerainbowFilter/SmoothUV/libsmoothuv.dll")
  20. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/GrainFilter/AddGrain/AddGrain.dll")
  21. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/NEO_FFT3DFilter/neo-fft3d.dll")
  22. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/DFTTest/DFTTest.dll")
  23. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/EEDI3m_opencl.dll") # vsQTGMC
  24. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/ResizeFilter/nnedi3/NNEDI3CL.dll")
  25. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools.dll")
  26. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/fmtconv.dll")
  27. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeinterlaceFilter/Bwdif/Bwdif.dll")
  28. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/akarin.dll")
  29. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/ZSmooth/zsmooth.dll")
  30. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SourceFilter/DGDecNV/DGDecodeNV_AVX2.dll")
  31. # Import scripts
  32. import masked
  33. import RainbowSmooth
  34. import qtgmc
  35. import validate
  36. # Source: 'C:\Users\Selur\Desktop\OPBox2.mkv'
  37. # Current color space: YUV420P8, bit depth: 8, resolution: 1920x1080, frame rate: 29.97fps, scanorder: top field first, yuv luminance scale: limited, matrix: 709, format: AVC
  38. # Loading 'C:\Users\Selur\Desktop\OPBox2.mkv' using DGSource
  39. clip = core.dgdecodenv.DGSource("J:/tmp/mkv_0604cae322f9fbe00040844b856ac3a4_853323747.dgi",fieldop=0) # 29.97 fps, scanorder: top field first
  40. frame = clip.get_frame(0)
  41. # setting color matrix to 709.
  42. clip = core.std.SetFrameProps(clip, _Matrix=vs.MATRIX_BT709)
  43. # setting color transfer (vs.TRANSFER_BT709), if it is not set.
  44. if validate.transferIsInvalid(clip):
  45. clip = core.std.SetFrameProps(clip=clip, _Transfer=vs.TRANSFER_BT709)
  46. # setting color primaries info (to vs.PRIMARIES_BT709), if it is not set.
  47. if validate.primariesIsInvalid(clip):
  48. clip = core.std.SetFrameProps(clip=clip, _Primaries=vs.PRIMARIES_BT709)
  49. # setting color range to TV (limited) range.
  50. prop_name = '_Range' if core.core_version.release_major >= 74 else '_ColorRange'
  51. clip = core.std.SetFrameProps(clip=clip, prop_name=vs.RANGE_LIMITED)
  52. # making sure frame rate is set to 29.97fps
  53. clip = core.std.AssumeFPS(clip=clip, fpsnum=30000, fpsden=1001)
  54. # making sure the detected scan type is set (detected: top field first)
  55. clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_TOP) # scan type: top field first
  56. # Deinterlacing using QTGMC
  57. clip = qtgmc.QTGMC(clip, Preset="Fast", TFF=True, opencl=True) # new fps: 59.94
  58. # Making sure content is preceived as frame based
  59. clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_PROGRESSIVE) # scan type: progressive
  60. clip = core.std.AddBorders(clip=clip, left=0, right=0, top=4, bottom=4) # add borders to archive mod 16 (vsRainbowSmooth) - 1920x1088
  61. # rainbow removal using RainbowSmooth
  62. clip = RainbowSmooth.RainbowSmooth(clip)
  63. clip = core.std.Crop(clip=clip, left=0, right=0, top=4, bottom=4) # removing added borders from mod requirement (vsRainbowSmooth) - 1920x1080
  64. # adjusting color space from YUV420P8 to RGBS for vsRealCUGAN
  65. clip = core.resize.Bicubic(clip=clip, format=vs.RGBS, matrix_in_s="709", range_in_s="limited", range_s="full", dither_type="error_diffusion")
  66. # resizing using RealCUGAN
  67. clip = core.rcnv.RealCUGAN(clip, noise=0, model=0, syncgap=2) # 3840x2160
  68. # resizing 3840x2160 to 1920x1080
  69. clip = core.fmtc.resample(clip, w=1920, h=1080, kernel="spline64", interlaced=False, interlacedd=False)
  70. clipMask = clip
  71. ## Starting applying 'BloatedEdgeMask' masked filtering for vsBasicVSRPPFilter
  72. # Creating mask for clip
  73. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.GRAY16, matrix_s="709", range_in_s="full", range_s="limited") # additional resize to match target color sampling
  74. clipMask = masked.bloated_edgemask(src=clipMask)
  75. for i in range(3):
  76. clipMask = core.std.Maximum(clipMask)
  77. clipFiltered = clip
  78. # adjusting color space from RGBS to RGBH for vsBasicVSRPPFilter
  79. clipFiltered = core.resize.Bicubic(clip=clipFiltered, format=vs.RGBH)
  80. # Quality enhancement using BasicVSR++
  81. from vsbasicvsrpp import basicvsrpp as BasicVSRPP
  82. clipFiltered = BasicVSRPP(clipFiltered, model=4)
  83. # adjusting clipFiltered color before mask merge
  84. clipFiltered = core.resize.Bicubic(clip=clipFiltered, format=vs.RGBS, dither_type="error_diffusion")
  85. # adjusting mask before mask merge
  86. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.GRAYS, range_in_s="limited", dither_type="error_diffusion")
  87. # Resolution: clip: 1920x1080 and clipFiltered: 1920x1080 and clipMask: 1920x1080
  88. # Merging: clip and clipFiltered based on clipMask
  89. clip = core.std.MaskedMerge(clip, clipFiltered, clipMask) # BloatedEdgeMask
  90. ## Finished applying 'BloatedEdgeMask' masked filtering for vsBasicVSRPPFilter
  91. # adjusting output color from RGBS to YUV420P10 for NVEncModel
  92. clip = core.resize.Bicubic(clip=clip, format=vs.YUV420P10, matrix_s="709", range_in_s="full", range_s="limited") # additional resize to match target color sampling
  93. # set output frame rate to 59.94fps (progressive)
  94. clip = core.std.AssumeFPS(clip=clip, fpsnum=60000, fpsden=1001)
  95. # output
  96. clip.set_output()
  97. # script was created by Hybrid 2026.04.06.1
Advertisement
Add Comment
Please, Sign In to add comment