Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # Imports
- import vapoursynth as vs
- # getting Vapoursynth core
- import ctypes
- import sys
- import os
- core = vs.core
- # Import scripts folder
- scriptPath = 'F:/Hybrid/64bit/vsscripts'
- sys.path.insert(0, os.path.abspath(scriptPath))
- # Loading Support Files
- Dllref = ctypes.windll.LoadLibrary("F:/Hybrid/64bit/vsfilters/Support/libfftw3f-3.dll")
- # loading plugins
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libvs_placebo.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SharpenFilter/AWarpSharp2/libawarpsharp2.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/TTempSmooth/TTempSmooth.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/DCTFilter.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/FFT3DFilter/fft3dfilter.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/CTMF/CTMF.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools_sf_em64t.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/TCanny.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SharpenFilter/CAS/CAS.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeblockFilter/DeblockPP7/DeblockPP7.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeblockFilter/Deblock/Deblock.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/ZSmooth/zsmooth.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/GrainFilter/RemoveGrain/RemoveGrainVS.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/GrainFilter/AddGrain/AddGrain.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/NEO_FFT3DFilter/neo-fft3d.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DenoiseFilter/DFTTest/DFTTest.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/EEDI3m_opencl.dll")# vsQTGMC
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/ResizeFilter/nnedi3/NNEDI3CL.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/libmvtools.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/Support/fmtconv.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/MiscFilter/MiscFilters/MiscFilters.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/DeinterlaceFilter/Bwdif/Bwdif.dll")
- core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SourceFilter/LSmashSource/LSMASHSource.dll")
- # Import scripts
- import edi_rpow2
- import dehalo
- import denoise
- import degrain
- import edge
- import masked
- import SpotLess
- import qtgmc
- import validate
- # Source: 'C:\Users\Selur\Desktop\BBCM.avi'
- # 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
- # Loading C:\Users\Selur\Desktop\BBCM.avi using LWLibavSource
- clip = core.lsmas.LWLibavSource(source="C:/Users/Selur/Desktop/BBCM.avi", format="YUV422P8", stream_index=0, cache=0, prefer_hw=0)
- frame = clip.get_frame(0)
- # setting color matrix to 470bg.
- clip = core.std.SetFrameProps(clip, _Matrix=vs.MATRIX_BT470_BG)
- # setting color transfer (vs.TRANSFER_BT601), if it is not set.
- if validate.transferIsInvalid(clip):
- clip = core.std.SetFrameProps(clip=clip, _Transfer=vs.TRANSFER_BT601)
- # setting color primaries info (to vs.PRIMARIES_BT470_BG), if it is not set.
- if validate.primariesIsInvalid(clip):
- clip = core.std.SetFrameProps(clip=clip, _Primaries=vs.PRIMARIES_BT470_BG)
- # setting color range to TV (limited) range.
- clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_LIMITED)
- # making sure frame rate is set to 29.97fps
- clip = core.std.AssumeFPS(clip=clip, fpsnum=30000, fpsden=1001)
- # making sure the detected scan type is set (detected: top field first)
- clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_TOP) # tff
- # Deinterlacing using QTGMC
- clip = qtgmc.QTGMC(Input=clip, Preset="Fast", TFF=True, opencl=True) # new fps: 59.94
- # Making sure content is preceived as frame based
- clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_PROGRESSIVE) # progressive
- # ColorMatrix: adjusting color matrix from 470bg to 2020ncl using transfer BT.709
- clip = core.std.SetFrameProps(clip=clip,_Transfer=1)
- # adjusting luma range to 'limited' due to post clipping
- clip = core.resize.Bicubic(clip=clip, matrix_in_s="470bg", matrix_s="2020ncl", range_in=0, range=0)
- # Spot removal using SpotLess
- clip = SpotLess.SpotLess(clip=clip, radT=3, pel=1, smoother="zsmooth")
- ## Starting applying 'limit' masked filtering for vsDeblock
- clipMask = clip
- clipMask = core.resize.Bicubic(clip=clipMask, format=vs.GRAY8, range_s="limited")
- clipMask = core.std.BinarizeMask(clipMask, 30)
- clipMask = core.std.InvertMask(clipMask)
- clipFiltered = clip
- # applying delocking using Deblock
- clipFiltered = core.deblock.Deblock(clip=clipFiltered, quant=40)
- clipMask = core.resize.Bicubic(clip=clipMask, format=vs.YUV422P8, range_s="limited")
- clip = core.std.MaskedMerge(clip, clipFiltered, clipMask) # LimitMask
- ## Finished applying 'LimitMask' masked filtering for vsDeblock
- # applying deblocking using DeblockPP7
- clip = core.pp7.DeblockPP7(clip=clip, qp=6.00)
- clip = core.std.Crop(clip=clip, left=8, right=12, top=2, bottom=4)# cropping to 700x474
- # Fix bright and dark line artifacts near the border of an image using BalanceBorders
- clip = edge.bbmod(c=clip,cLeft=4,cTop=0,cRight=12,cBottom=8)
- # contrast sharpening using CAS
- clip = core.cas.CAS(clip=clip)
- # removing grain using TemporalDegrain2
- clip = degrain.TemporalDegrain2(clip=clip, degrainPlane=4, meAlgPar=False, postFFT=0, fftThreads=1)
- clipMerge = clip
- clipMerge = core.std.AddBorders(clip=clipMerge, left=2, right=2, top=2, bottom=4) # add borders to archive mod 8 (vsBasicVSRPPFilter) - 704x480
- # adjusting color space from YUV422P8 to RGBH for vsBasicVSRPPFilter
- clipMerge = core.resize.Bicubic(clip=clipMerge, format=vs.RGBH, matrix_in_s="2020ncl", range_s="limited")
- # Quality enhancement using BasicVSR++
- from vsbasicvsrpp import basicvsrpp as BasicVSRPP
- clipMerge = BasicVSRPP(clip=clipMerge, model=4)
- clipMerge = core.std.Crop(clip=clipMerge, left=2, right=2, top=2, bottom=4) # removing borders (vsBasicVSRPPFilter) - 700x474
- clipMerge = core.resize.Bicubic(clip=clipMerge, format=vs.YUV444P16, matrix_s="2020ncl", range_s="limited")
- clip = core.resize.Bicubic(clip=clip, format=vs.YUV444P16, range_s="limited")
- clip = core.std.Merge(clip,clipMerge,[0, 1, 1]) # merge filtered and unfiltered by plane
- clip = core.std.AddBorders(clip=clip, left=0, right=0, top=0, bottom=2) # add borders to archive mod 4 (vsMCTemporalDenoise) - 700x476
- # denoising using MCTemporalDenoise
- clip = denoise.MCTemporalDenoise(i=clip, settings="medium", ncpu=1)
- clip = core.std.Crop(clip=clip, left=0, right=0, top=0, bottom=2) # removing borders (vsMCTemporalDenoise) - 700x474
- # applying dehalo using YAHR
- clip = dehalo.YAHR(clip)
- clip = core.std.AddBorders(clip=clip, left=0, right=0, top=0, bottom=2) # add borders to archive mod 4 (NNEDI3(CL)) - 700x476
- # resizing using NNEDI3CL
- # current: 700x476 target: 1440x1098 -> pow: 4
- clip = edi_rpow2.nnedi3cl_rpow2(clip=clip, rfactor=4, nns=4) # 2800x1904
- clip = core.std.Crop(clip=clip, left=0, right=0, top=0, bottom=8) # removing borders (NNEDI3(CL)) - 2800x1896
- # resizing 2800x1896 to 1440x1098
- clip = core.fmtc.resample(clip=clip, w=1440, h=1098, kernel="spline64", interlaced=False, interlacedd=False)# before YUV444P16 after YUV444P16
- with open("F:/Hybrid/64bit/vsfilters/GLSL/parameterized/CAS.glsl") as glslf:
- glsl = glslf.read()
- glsl = glsl.replace('#define SHARPENING 0.0', '#define SHARPENING 0')
- glsl = glsl.replace('#define CAS_BETTER_DIAGONALS 1', '#define CAS_BETTER_DIAGONALS 1')
- glsl = glsl.replace('#define CAS_GO_SLOWER 0', '#define CAS_GO_SLOWER 0')
- glsl = glsl.replace('#define SOURCE_TRC 0', '#define SOURCE_TRC 0')
- glsl = glsl.replace('#define TARGET_TRC 0', '#define TARGET_TRC 0')
- clip = core.placebo.Shader(clip=clip, shader_s=glsl, width=clip.width, height=clip.height)
- # adjusting output color from: YUV444P16 to YUV420P10 for NVEncModel
- clip = core.resize.Bicubic(clip=clip, format=vs.YUV420P10, range_s="limited", dither_type="error_diffusion")
- # set output frame rate to 59.94fps (progressive)
- clip = core.std.AssumeFPS(clip=clip, fpsnum=60000, fpsden=1001)
- # output
- clip.set_output()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement