DPIR (denoiser) - Printable Version

# Imports

import vapoursynth as vs

# getting Vapoursynth core

import ctypes

import site

import sys

import os

core = vs.core

# Import scripts folder

scriptPath = 'F:/Hybrid/64bit/vsscripts'

sys.path.insert(0, os.path.abspath(scriptPath))

# Adding torch dependencies to PATH

path = site.getsitepackages()[0]+'/torch_dependencies/bin/'

ctypes.windll.kernel32.SetDllDirectoryW(path)

path = path.replace('\\', '/')

os.environ["PATH"] = path + os.pathsep + os.environ["PATH"]

# Loading Plugins

core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/SourceFilter/LSmashSource/vslsmashsource.dll")

# source: 'C:\Users\Selur\Desktop\Source\AKG01.mkv'

# current color space: YUV420P10, bit depth: 10, resolution: 1920x1080, fps: 23.976, scanorder: progressive, yuv luminance scale: limited, matrix: 709, transfer: bt.709, primaries: bt.709

# Loading C:\Users\Selur\Desktop\Source\AKG01.mkv using LWLibavSource

clip = core.lsmas.LWLibavSource(source="C:/Users/Selur/Desktop/Source/AKG01.mkv", format="YUV420P10", stream_index=0, cache=0)

frame = clip.get_frame(0)

# Setting detected color matrix (709).

clip = core.std.SetFrameProps(clip, _Matrix=1)

# Setting color transfer (709), if it is not set.

if '_Transfer' not in frame.props or not frame.props['_Transfer']:

  clip = core.std.SetFrameProps(clip, _Transfer=1)

# Setting color primaries info (to 709), if it is not set.

if '_Primaries' not in frame.props or not frame.props['_Primaries']:

  clip = core.std.SetFrameProps(clip, _Primaries=1)

# Setting color range to TV (limited) range.

clip = core.std.SetFrameProp(clip=clip, prop="_ColorRange", intval=1)

# making sure frame rate is set to 23.976

clip = core.std.AssumeFPS(clip=clip, fpsnum=24000, fpsden=1001)

clip = core.std.SetFrameProp(clip=clip, prop="_FieldBased", intval=0) # progressive

from vsdpir import dpir as DPIR

# adjusting color space from YUV420P10 to RGBH for vsDPIRDenoise

clip = core.resize.Bicubic(clip=clip, format=vs.RGBH, matrix_in_s="709", range_s="limited")

# denoising using DPIRDenoise

clip = DPIR(clip=clip, strength=5.000, task="denoise", device_index=0)

# adjusting output color from: RGBH to YUV420P10 for NVEncModel

clip = core.resize.Bicubic(clip=clip, format=vs.YUV420P10, matrix_s="709", range_s="limited", dither_type="error_diffusion")

# set output frame rate to 23.976fps (progressive)

clip = core.std.AssumeFPS(clip=clip, fpsnum=24000, fpsden=1001)

# Output

clip.set_output()