Last active
December 26, 2023 19:33
-
-
Save YamashitaRen/020c497524e794779d9c to your computer and use it in GitHub Desktop.
Put it in Python\Lib\site-packages. Usage : import edi_rpow2 as edi \ edi.nnedi3_rpow2(clip,rfactor,...) \ edi.eedi3_rpow2(clip,rfactor,...) \ edi.eedi2_rpow2(clip,rfactor,...)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
def nnedi3_rpow2(clip,rfactor,correct_shift="zimg",nsize=0,nns=3,qual=None,etype=None,pscrn=None,opt=None,int16_prescreener=None,int16_predictor=None,exp=None): | |
import vapoursynth as vs | |
core = vs.get_core() | |
def edi(clip,field,dh): | |
return core.nnedi3.nnedi3(clip=clip,field=field,dh=dh,nsize=nsize,nns=nns,qual=qual,etype=etype,pscrn=pscrn,opt=opt,int16_prescreener=int16_prescreener,int16_predictor=int16_predictor,exp=exp) | |
return edi_rpow2(clip=clip,rfactor=rfactor,correct_shift=correct_shift,edi=edi) | |
def znedi3_rpow2(clip,rfactor,correct_shift="zimg",nsize=0,nns=3,qual=None,etype=None,pscrn=None,opt=None,int16_prescreener=None,int16_predictor=None,exp=None): | |
import vapoursynth as vs | |
core = vs.get_core() | |
def edi(clip,field,dh): | |
return core.znedi3.nnedi3(clip=clip,field=field,dh=dh,nsize=nsize,nns=nns,qual=qual,etype=etype,pscrn=pscrn,opt=opt,int16_prescreener=int16_prescreener,int16_predictor=int16_predictor,exp=exp) | |
return edi_rpow2(clip=clip,rfactor=rfactor,correct_shift=correct_shift,edi=edi) | |
def nnedi3cl_rpow2(clip,rfactor,correct_shift="zimg",nsize=0,nns=3,qual=None,etype=None,pscrn=None,device=None): | |
import vapoursynth as vs | |
core = vs.get_core() | |
def edi(clip,field,dh): | |
return core.nnedi3cl.NNEDI3CL(clip=clip,field=field,dh=dh,nsize=nsize,nns=nns,qual=qual,etype=etype,pscrn=pscrn,device=device) | |
return edi_rpow2(clip=clip,rfactor=rfactor,correct_shift=correct_shift,edi=edi) | |
def eedi3_rpow2(clip,rfactor,correct_shift="zimg",alpha=None,beta=None,gamma=None,nrad=None,mdis=None,hp=None,ucubic=None,cost3=None,vcheck=None,vthresh0=None,vthresh1=None,vthresh2=None,sclip=None): | |
import vapoursynth as vs | |
core = vs.get_core() | |
def edi(clip,field,dh): | |
return core.eedi3.eedi3(clip=clip,field=field,dh=dh,alpha=alpha,beta=beta,gamma=gamma,nrad=nrad,mdis=mdis,hp=hp,ucubic=ucubic,cost3=cost3,vcheck=vcheck,vthresh0=vthresh0,vthresh1=vthresh1,vthresh2=vthresh2,sclip=sclip) | |
return edi_rpow2(clip=clip,rfactor=rfactor,correct_shift=correct_shift,edi=edi) | |
def eedi2_rpow2(clip,rfactor,correct_shift="zimg",mthresh=None,lthresh=None,vthresh=None,estr=None,dstr=None,maxd=None,map=None,nt=None,pp=None): | |
import vapoursynth as vs | |
core = vs.get_core() | |
def edi(clip,field,dh): | |
return core.eedi2.EEDI2(clip=clip,field=field,mthresh=mthresh,lthresh=lthresh,vthresh=vthresh,estr=estr,dstr=dstr,maxd=maxd,map=map,nt=nt,pp=pp) | |
return edi_rpow2(clip=clip,rfactor=rfactor,correct_shift=correct_shift,edi=edi) | |
def edi_rpow2(clip,rfactor,correct_shift,edi): | |
import vapoursynth as vs | |
import math | |
core = vs.get_core() | |
steps=math.log(rfactor)/math.log(2) # 2^steps=rfactor | |
if (steps).is_integer: | |
steps=int(steps) | |
else : | |
raise ValueError('nnedi3_rpow2 : rfactor must be a power of two') | |
if correct_shift not in [None,"zimg","fmtconv"]: | |
raise ValueError('correct_shift must be None, "zimg" or "fmtconv" ') | |
for i in range(0,2): | |
clip=core.std.Transpose(clip) | |
clip=edi(clip,field=1,dh=1) | |
for i in range(0,steps-1): | |
clip=core.std.Transpose(clip) | |
clip=edi(clip,field=1,dh=1) | |
clip=core.std.Transpose(clip) | |
clip=edi(clip,field=0,dh=1) | |
if correct_shift=="zimg" or correct_shift=="fmtconv": | |
clip=correct_edi_shift(clip,rfactor=rfactor,plugin=correct_shift) | |
return clip | |
def correct_edi_shift(clip,rfactor,plugin): | |
import vapoursynth as vs | |
core = vs.get_core() | |
if clip.format.subsampling_w==1: | |
hshift=-rfactor/2+0.5 # hshift(steps+1)=2*hshift(steps)-0.5 | |
else : | |
hshift=-0.5 | |
if plugin=="zimg": | |
if clip.format.subsampling_h==0: | |
clip=core.resize.Spline36(clip=clip,width=clip.width,height=clip.height,src_left=hshift,src_top=-0.5) | |
else : | |
Y=core.std.ShufflePlanes(clips=clip, planes=0, colorfamily=vs.GRAY) | |
U=core.std.ShufflePlanes(clips=clip, planes=1, colorfamily=vs.GRAY) | |
V=core.std.ShufflePlanes(clips=clip, planes=2, colorfamily=vs.GRAY) | |
Y=core.resize.Spline36(clip=Y,width=clip.width,height=clip.height,src_left=hshift,src_top=-0.5) | |
U=core.resize.Spline36(clip=U,width=clip.width,height=clip.height,src_left=hshift/2,src_top=-0.5) | |
V=core.resize.Spline36(clip=V,width=clip.width,height=clip.height,src_left=hshift/2,src_top=-0.5) | |
clip=core.std.ShufflePlanes(clips=[Y,U,V], planes=[0,0,0], colorfamily=vs.YUV) | |
if plugin=="fmtconv": | |
bits=clip.format.bits_per_sample | |
if clip.format.subsampling_h==0: | |
clip=core.fmtc.resample(clip=clip,sx=hshift,sy=-0.5) | |
else : | |
clip=core.fmtc.resample(clip=clip,sx=hshift,sy=-0.5,planes=[3,2,2]) | |
clip=core.fmtc.resample(clip=clip,sx=hshift,sy=-1,planes=[2,3,3]) | |
if bits!=16: | |
clip=core.fmtc.bitdepth(clip=clip,bits=bits) | |
return clip |
@Selur Thanks !
I totally missed it. Was thinking about updating the script for nnedi3_cl but you're forcing my hand ;)
And while you're at it, why not use the faster znedi instead of nnedi. See also here for a nnedicl + znedi "switch": https://github.com/HomeOfVapourSynthEvolution/havsfunc/blob/master/havsfunc.py#L81
Yeah that's in plan too @theChaosCoder. Starting now °°
Done. NNEDI3CL isn't much tested since I get artifacts on the output and/or GPU crash.
This doesn’t work with VapourSynth R55+ because it removed the deprecated vs.get_core()
(vs.core
should be used instead). Gist with the fix applied
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
small addition to add support for NNEDI3CL (https://github.com/HomeOfVapourSynthEvolution/VapourSynth-NNEDI3CL)