Guest User

slow

a guest
Jul 13th, 2025
22
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 5.14 KB | None | 0 0
  1. # Imports
  2. import vapoursynth as vs
  3. # getting Vapoursynth core
  4. import sys
  5. import os
  6. core = vs.core
  7. # Import scripts folder
  8. scriptPath = 'F:/Hybrid/64bit/vsscripts'
  9. sys.path.insert(0, os.path.abspath(scriptPath))
  10. # loading plugins
  11. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/ZSmooth/zsmooth.dll")
  12. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/CTMF/CTMF.dll")
  13. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools.dll")
  14. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/vszip.dll")
  15. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/akarin.dll")
  16. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/fmtconv.dll")
  17. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SourceFilter/BestSource/BestSource.dll")
  18. # Import scripts
  19. import nnedi3_resample
  20. import smdegrain
  21. import animate
  22. import masked
  23. import validate
  24. # Source: 'C:\Users\Selur\Desktop\testfile.mp4'
  25. # Current color space: YUV420P8, bit depth: 8, resolution: 1920x1080, frame rate: 23.976fps, scanorder: progressive, yuv luminance scale: limited, matrix: 709, transfer: bt.709, primaries: bt.709, format: AVC
  26. # Loading C:\Users\Selur\Desktop\testfile.mp4 using BestSource)
  27. clip = core.bs.VideoSource(source="C:/Users/Selur/Desktop/testfile.mp4", cachepath="J:/tmp/testfile_bestSource", track=0, hwdevice="opencl")
  28. frame = clip.get_frame(0)
  29. # setting color matrix to 709.
  30. clip = core.std.SetFrameProps(clip, _Matrix=vs.MATRIX_BT709)
  31. # setting color transfer (vs.TRANSFER_BT709), if it is not set.
  32. if validate.transferIsInvalid(clip):
  33. clip = core.std.SetFrameProps(clip=clip, _Transfer=vs.TRANSFER_BT709)
  34. # setting color primaries info (to vs.PRIMARIES_BT709), if it is not set.
  35. if validate.primariesIsInvalid(clip):
  36. clip = core.std.SetFrameProps(clip=clip, _Primaries=vs.PRIMARIES_BT709)
  37. # setting color range to TV (limited) range.
  38. clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_LIMITED)
  39. # making sure frame rate is set to 23.976fps
  40. clip = core.std.AssumeFPS(clip=clip, fpsnum=24000, fpsden=1001)
  41. # making sure the detected scan type is set (detected: progressive)
  42. clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_PROGRESSIVE) # progressive
  43. clip = core.fmtc.resample(clip=clip, w=960, h=480, kernel="cubic", interlaced=False, interlacedd=False) # resizing for 'Gimmick' (vsBasicVSRPPFilter)
  44. # adjusting color space from YUV420P16 to RGBH for vsBasicVSRPPFilter
  45. clip = core.resize.Bicubic(clip=clip, format=vs.RGBH, matrix_in_s="709", range_s="limited")
  46. ## Starting applying 'CartoonEdges' masked filtering for vsBasicVSRPPFilter
  47. clipMask = clip
  48. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.YUV444P16, matrix_s="709", range_s="limited")
  49. clipMask = masked.CartoonEdges(clip=clipMask, low=19, high=255)
  50. for i in range(2):
  51. clipMask = core.std.Maximum(clipMask)
  52. clipMask = core.std.Expr(clipMask, expr=f"x 1.7 *")
  53. clipMask = core.std.BinarizeMask(clip=clipMask,threshold=6016)
  54. clipFiltered = clip
  55. # Quality enhancement using BasicVSR++
  56. from vsbasicvsrpp import basicvsrpp as BasicVSRPP
  57. clipFiltered = core.resize.Bicubic(clip=clipFiltered, format=vs.RGBS, range_s="limited")
  58. def vsBasicVSRPPFilterRegion1(clipFiltered,*args):
  59. return BasicVSRPP(clip=clipFiltered, model=4)
  60. end = clipFiltered.num_frames-1
  61. mapping = [
  62. (0, end), [vsBasicVSRPPFilterRegion1],
  63. ]
  64. clipFiltered = animate.run(clipFiltered, mapping, selection=(480,256,150,100))
  65. clipFiltered = core.resize.Bicubic(clip=clipFiltered, format=vs.RGBH, range_s="limited", dither_type="error_diffusion")
  66. clipFiltered = core.resize.Bicubic(clip=clipFiltered, format=vs.RGBS, range_s="limited")
  67. clipFiltered = core.fmtc.resample(clip=clipFiltered, w=1920, h=1080, kernel="cubic", interlaced=False, interlacedd=False) # undo resizing for 'Gimmick' (vsBasicVSRPPFilter)
  68. clipMask = core.std.Expr(clipMask, expr=f"x 1.7 *")
  69. clipMask = core.std.BinarizeMask(clip=clipMask,threshold=6016)
  70. clip = core.resize.Bicubic(clip=clip, format=vs.RGBS, range_s="limited")
  71. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.GRAYS, range_s="limited")
  72. clipMask = core.resize.Spline64(clip=clipMask, width=1920, height=1080)# resize to match filtered dimensions
  73. clip = core.resize.Spline64(clip=clip, width=1920, height=1080)# resize to match filtered dimensions
  74. clip = core.std.MaskedMerge(clip, clipFiltered, clipMask) # CartoonEdges
  75. ## Finished applying 'CartoonEdges' masked filtering for vsBasicVSRPPFilter
  76. # making sure 0-1 limits are respected
  77. clip = core.vszip.Limiter(clip=clip, min=[0,0,0], max=[1,1,1])
  78. # adjusting color space from RGBS to YUV444P16 for vsSMDegrain
  79. clip = core.resize.Bicubic(clip=clip, format=vs.YUV444P16, matrix_s="709", range_s="limited", dither_type="error_diffusion")
  80. # removing grain using SMDegrain
  81. clip = smdegrain.SMDegrain(input=clip, contrasharp=50, interlaced=False, dct=1, opencl=True, device=-1)
  82. # adjusting output color from: YUV444P16 to YUV420P10 for NVEncModel
  83. clip = core.resize.Bicubic(clip=clip, format=vs.YUV420P10, range_s="limited", dither_type="error_diffusion")
  84. # set output frame rate to 23.976fps (progressive)
  85. clip = core.std.AssumeFPS(clip=clip, fpsnum=24000, fpsden=1001)
  86. # output
  87. clip.set_output()
Advertisement
Add Comment
Please, Sign In to add comment