diff --git a/lostfunc.py b/lostfunc.py index d4587e5..f954cb7 100644 --- a/lostfunc.py +++ b/lostfunc.py @@ -1,6 +1,6 @@ from vapoursynth import core, YUV, GRAY import vapoursynth as vs -import adjust +import adjust, havsfunc as haf import math @@ -59,8 +59,8 @@ def Stab(clp, range=1, dxmax=4, dymax=4, mirror=0): if not isinstance(clp, vs.VideoNode): raise TypeError('Stab: This is not a clip') - temp = AverageFrames(clp, weights=[1] * 15, scenechange=25 / 255) - inter = core.std.Interleave([core.rgvs.Repair(temp, AverageFrames(clp, weights=[1] * 3, scenechange=25 / 255), 1), clp]) + temp = haf.AverageFrames(clp, weights=[1] * 15, scenechange=25 / 255) + inter = core.std.Interleave([core.rgvs.Repair(temp, haf.AverageFrames(clp, weights=[1] * 3, scenechange=25 / 255), 1), clp]) mdata = core.depan.DePanEstimate(inter, range=range, trust=0, dxmax=dxmax, dymax=dymax) last = core.depan.DePan(inter, data=mdata, offset=-1, mirror=mirror) return last[::2]