Advertisement
Guest User

conventional filtering

a guest
Mar 10th, 2025
41
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 8.20 KB | None | 0 0
  1. # Imports
  2. import vapoursynth as vs
  3. # getting Vapoursynth core
  4. import ctypes
  5. import sys
  6. import os
  7. core = vs.core
  8. # Import scripts folder
  9. scriptPath = 'F:/Hybrid/64bit/vsscripts'
  10. sys.path.insert(0, os.path.abspath(scriptPath))
  11. # Loading Support Files
  12. Dllref = ctypes.windll.LoadLibrary("F:/Hybrid/64bit/vsfilters/Support/libfftw3f-3.dll")
  13. # loading plugins
  14. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libvs_placebo.dll")
  15. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SharpenFilter/AWarpSharp2/libawarpsharp2.dll")
  16. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/TTempSmooth/TTempSmooth.dll")
  17. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/DCTFilter.dll")
  18. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/FFT3DFilter/fft3dfilter.dll")
  19. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/CTMF/CTMF.dll")
  20. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools_sf_em64t.dll")
  21. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/TCanny.dll")
  22. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SharpenFilter/CAS/CAS.dll")
  23. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeblockFilter/DeblockPP7/DeblockPP7.dll")
  24. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeblockFilter/Deblock/Deblock.dll")
  25. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/ZSmooth/zsmooth.dll")
  26. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/GrainFilter/RemoveGrain/RemoveGrainVS.dll")
  27. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/GrainFilter/AddGrain/AddGrain.dll")
  28. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/NEO_FFT3DFilter/neo-fft3d.dll")
  29. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/DFTTest/DFTTest.dll")
  30. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/EEDI3m_opencl.dll")# vsQTGMC
  31. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/ResizeFilter/nnedi3/NNEDI3CL.dll")
  32. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools.dll")
  33. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/fmtconv.dll")
  34. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/MiscFilter/MiscFilters/MiscFilters.dll")
  35. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeinterlaceFilter/Bwdif/Bwdif.dll")
  36. core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SourceFilter/LSmashSource/LSMASHSource.dll")
  37. # Import scripts
  38. import edi_rpow2
  39. import dehalo
  40. import denoise
  41. import degrain
  42. import edge
  43. import masked
  44. import SpotLess
  45. import qtgmc
  46. import validate
  47. # Source: 'C:\Users\Selur\Desktop\BBCM.avi'
  48. # Current color space: YUV422P8, bit depth: 8, resolution: 720x480, frame rate: 29.97fps, scanorder: top field first, yuv luminance scale: limited, matrix: 470bg, format: Lagarith
  49. # Loading C:\Users\Selur\Desktop\BBCM.avi using LWLibavSource
  50. clip = core.lsmas.LWLibavSource(source="C:/Users/Selur/Desktop/BBCM.avi", format="YUV422P8", stream_index=0, cache=0, prefer_hw=0)
  51. frame = clip.get_frame(0)
  52. # setting color matrix to 470bg.
  53. clip = core.std.SetFrameProps(clip, _Matrix=vs.MATRIX_BT470_BG)
  54. # setting color transfer (vs.TRANSFER_BT601), if it is not set.
  55. if validate.transferIsInvalid(clip):
  56. clip = core.std.SetFrameProps(clip=clip, _Transfer=vs.TRANSFER_BT601)
  57. # setting color primaries info (to vs.PRIMARIES_BT470_BG), if it is not set.
  58. if validate.primariesIsInvalid(clip):
  59. clip = core.std.SetFrameProps(clip=clip, _Primaries=vs.PRIMARIES_BT470_BG)
  60. # setting color range to TV (limited) range.
  61. clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_LIMITED)
  62. # making sure frame rate is set to 29.97fps
  63. clip = core.std.AssumeFPS(clip=clip, fpsnum=30000, fpsden=1001)
  64. # making sure the detected scan type is set (detected: top field first)
  65. clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_TOP) # tff
  66. # Deinterlacing using QTGMC
  67. clip = qtgmc.QTGMC(Input=clip, Preset="Fast", TFF=True, opencl=True) # new fps: 59.94
  68. # Making sure content is preceived as frame based
  69. clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_PROGRESSIVE) # progressive
  70. # ColorMatrix: adjusting color matrix from 470bg to 2020ncl using transfer BT.709
  71. clip = core.std.SetFrameProps(clip=clip,_Transfer=1)
  72. # adjusting luma range to 'limited' due to post clipping
  73. clip = core.resize.Bicubic(clip=clip, matrix_in_s="470bg", matrix_s="2020ncl", range_in=0, range=0)
  74. # Spot removal using SpotLess
  75. clip = SpotLess.SpotLess(clip=clip, radT=3, pel=1, smoother="zsmooth")
  76. ## Starting applying 'limit' masked filtering for vsDeblock
  77. clipMask = clip
  78. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.GRAY8, range_s="limited")
  79. clipMask = core.std.BinarizeMask(clipMask, 30)
  80. clipMask = core.std.InvertMask(clipMask)
  81. clipFiltered = clip
  82. # applying delocking using Deblock
  83. clipFiltered = core.deblock.Deblock(clip=clipFiltered, quant=40)
  84. clipMask = core.resize.Bicubic(clip=clipMask, format=vs.YUV422P8, range_s="limited")
  85. clip = core.std.MaskedMerge(clip, clipFiltered, clipMask) # LimitMask
  86. ## Finished applying 'LimitMask' masked filtering for vsDeblock
  87. # applying deblocking using DeblockPP7
  88. clip = core.pp7.DeblockPP7(clip=clip, qp=6.00)
  89. clip = core.std.Crop(clip=clip, left=8, right=12, top=2, bottom=4)# cropping to 700x474
  90. # Fix bright and dark line artifacts near the border of an image using BalanceBorders
  91. clip = edge.bbmod(c=clip,cLeft=4,cTop=0,cRight=12,cBottom=8)
  92. # contrast sharpening using CAS
  93. clip = core.cas.CAS(clip=clip)
  94. # removing grain using TemporalDegrain2
  95. clip = degrain.TemporalDegrain2(clip=clip, degrainPlane=4, meAlgPar=False, postFFT=0, fftThreads=1)
  96. clipMerge = clip
  97. clipMerge = core.std.AddBorders(clip=clipMerge, left=2, right=2, top=2, bottom=4) # add borders to archive mod 8 (vsBasicVSRPPFilter) - 704x480
  98. # adjusting color space from YUV422P8 to RGBH for vsBasicVSRPPFilter
  99. clipMerge = core.resize.Bicubic(clip=clipMerge, format=vs.RGBH, matrix_in_s="2020ncl", range_s="limited")
  100. # Quality enhancement using BasicVSR++
  101. from vsbasicvsrpp import basicvsrpp as BasicVSRPP
  102. clipMerge = BasicVSRPP(clip=clipMerge, model=4)
  103. clipMerge = core.std.Crop(clip=clipMerge, left=2, right=2, top=2, bottom=4) # removing borders (vsBasicVSRPPFilter) - 700x474
  104. clipMerge = core.resize.Bicubic(clip=clipMerge, format=vs.YUV444P16, matrix_s="2020ncl", range_s="limited")
  105. clip = core.resize.Bicubic(clip=clip, format=vs.YUV444P16, range_s="limited")
  106. clip = core.std.Merge(clip,clipMerge,[0, 1, 1]) # merge filtered and unfiltered by plane
  107. clip = core.std.AddBorders(clip=clip, left=0, right=0, top=0, bottom=2) # add borders to archive mod 4 (vsMCTemporalDenoise) - 700x476
  108. # denoising using MCTemporalDenoise
  109. clip = denoise.MCTemporalDenoise(i=clip, settings="medium", ncpu=1)
  110. clip = core.std.Crop(clip=clip, left=0, right=0, top=0, bottom=2) # removing borders (vsMCTemporalDenoise) - 700x474
  111. # applying dehalo using YAHR
  112. clip = dehalo.YAHR(clip)
  113. clip = core.std.AddBorders(clip=clip, left=0, right=0, top=0, bottom=2) # add borders to archive mod 4 (NNEDI3(CL)) - 700x476
  114. # resizing using NNEDI3CL
  115. # current: 700x476 target: 1440x1098 -> pow: 4
  116. clip = edi_rpow2.nnedi3cl_rpow2(clip=clip, rfactor=4, nns=4) # 2800x1904
  117. clip = core.std.Crop(clip=clip, left=0, right=0, top=0, bottom=8) # removing borders (NNEDI3(CL)) - 2800x1896
  118. # resizing 2800x1896 to 1440x1098
  119. clip = core.fmtc.resample(clip=clip, w=1440, h=1098, kernel="spline64", interlaced=False, interlacedd=False)# before YUV444P16 after YUV444P16
  120. with open("F:/Hybrid/64bit/vsfilters/GLSL/parameterized/CAS.glsl") as glslf:
  121. glsl = glslf.read()
  122. glsl = glsl.replace('#define SHARPENING 0.0', '#define SHARPENING 0')
  123. glsl = glsl.replace('#define CAS_BETTER_DIAGONALS 1', '#define CAS_BETTER_DIAGONALS 1')
  124. glsl = glsl.replace('#define CAS_GO_SLOWER 0', '#define CAS_GO_SLOWER 0')
  125. glsl = glsl.replace('#define SOURCE_TRC 0', '#define SOURCE_TRC 0')
  126. glsl = glsl.replace('#define TARGET_TRC 0', '#define TARGET_TRC 0')
  127. clip = core.placebo.Shader(clip=clip, shader_s=glsl, width=clip.width, height=clip.height)
  128. # adjusting output color from: YUV444P16 to YUV420P10 for NVEncModel
  129. clip = core.resize.Bicubic(clip=clip, format=vs.YUV420P10, range_s="limited", dither_type="error_diffusion")
  130. # set output frame rate to 59.94fps (progressive)
  131. clip = core.std.AssumeFPS(clip=clip, fpsnum=60000, fpsden=1001)
  132. # output
  133. clip.set_output()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement