Deoldify Vapoursynth filter - Printable Version

def clip_color_stabilizer(clip: vs.VideoNode = None, nframes: int = 5, smooth_type: int = 0) -> vs.VideoNode:

    max_frames = max(1, min(nframes, 31))

        def smooth_frame(n, f):

        f_out = f.copy()

        if n < max_frames:

            return f_out

        img_f = list()

        img_f.append(frame_to_image(f)) 

        for i in range(1, max_frames):        

            img_f.append(frame_to_image(clip.get_frame(n-i)))

        img_m = color_temporal_stabilizer(img_f, max_frames)

        return image_to_frame(img_m, f_out)    

    clip = clip.std.ModifyFrame(clips=[clip], selector=smooth_frame)

    return clip



#-----------------------------------------------------------------------

def color_temporal_stabilizer(img_f: list, nframes: int = 5) -> Image:

    

    img_new = np.copy(np.asarray(img_f[0]))

    

    yuv_new = cv2.cvtColor(img_new, cv2.COLOR_RGB2YUV)

    

    weight: float = 1.0/nframes



    yuv_m = np.multiply(yuv_new, weight).clip(0, 255).astype(int)

    

    for i in range (1, nframes):

        yuv_i = cv2.cvtColor(np.asarray(img_f[i]), cv2.COLOR_RGB2YUV) 

        yuv_m += np.multiply(yuv_i, weight).clip(0, 255).astype(int)  

    

    yuv_new[:, :, 1] = yuv_m[:, :, 1]

    yuv_new[:, :, 2] = yuv_m[:, :, 2]

            

    return Image.fromarray(cv2.cvtColor(yuv_new, cv2.COLOR_YUV2RGB))

core.std.LoadPlugin(path="F:/Hybrid/64bit/vsfilters/MiscFilter/MiscFilters/MiscFilters.dll")

import math

def vs_clip_color_stabilizer(clip: vs.VideoNode = None, nframes: int = 5, mode: str = 'center', scenechange: bool = True) -> vs.VideoNode:

    

    if nframes < 3 or nframes > 31: 

        raise ValueError("deoldify: number of frames must be in range: 3-31")

        

    if mode not in ['left', 'center', 'right']:

        raise ValueError("deoldify: mode must be 'left', 'center', or 'right'.")

    

    # convert the clip format for AverageFrames to YUV    

    clip_yuv = clip.resize.Bicubic(format=vs.YUV444PS, matrix_s="709", range_s="limited")

    

    if nframes%2==0:

        nframes +=1

    

    N = max(3, min(nframes, 31))

    Nh = round((N-1)/2)    

    Wi = math.trunc(100.0/N)

    

    if mode in ['left', 'right']:

        Wi = 2*Wi        

        Wc = 100-(Nh)*Wi

    else:

        Wc = 100-(N-1)*Wi

    

    weight_list = list()       

    for i in range(0, Nh):

        if mode in ['left', 'center']:

            weight_list.append(Wi)

        else:

            weight_list.append(0)

    weight_list.append(Wc)    

    for i in range(0, Nh):

        if mode in ['right', 'center']:    

            weight_list.append(Wi)

        else:

            weight_list.append(0)

    

    clip_yuv = vs.core.misc.AverageFrames(clip_yuv, weight_list, scenechange = True, planes=[1,2])           

    

    # convert the clip format for deoldify to RGB24 

    clip_rgb = clip_yuv.resize.Bicubic(format=vs.RGB24, matrix_in_s="709", range_s="limited", dither_type="error_diffusion") 

    

    return clip_rgb

    

clip = vs_clip_color_stabilizer(clip, nframes = 31)