Deoldify Vapoursynth filter

[Selur]

@Dan64: A few questions:
a. HAVC_restore_video is not used, is that correct?
b. is it correct that HAVC_stabilize only gets used with deepEx?
Please update the graph accordingly.

How should ScFrameDir be set with method 5 (for a non-custom preset/speed)?

-----------------
The following code would be generated atm:

Default for a 470gb YUV source:

# changing color matrix from '470bg' to '709' for vsDeOldify
clip = core.resize.Bicubic(clip, matrix_in_s="470bg", matrix_s="709")
# changing range from limited to full range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="limited", range_s="full")
# setting color range to PC (full) range.
clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)
# adjusting color space from YUV420P8 to RGB24 for vsDeOldify
clip = core.resize.Bicubic(clip=clip, format=vs.RGB24, matrix_in_s="709", range_s="full")
# adding colors using DeOldify
from vsdeoldify import HAVC_main
clip = HAVC_main(clip=clip)
# internally changing color matrix for YUV<>RGB to '470bg' undoing color matrix change for vsDeOldify
# changing range from full to limited range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="full", range_s="limited")

Switching B&W Tune to 'light':

# changing color matrix from '470bg' to '709' for vsDeOldify
clip = core.resize.Bicubic(clip, matrix_in_s="470bg", matrix_s="709")
# changing range from limited to full range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="limited", range_s="full")
# setting color range to PC (full) range.
clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)
# adjusting color space from YUV420P8 to RGB24 for vsDeOldify
clip = core.resize.Bicubic(clip=clip, format=vs.RGB24, matrix_in_s="709", range_s="full")
# adding colors using DeOldify
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="ON")
from vsdeoldify import HAVC_main
clip = HAVC_main(clip=clip, BlackWhiteTune="light")
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="OFF")
# internally changing color matrix for YUV<>RGB to '470bg' undoing color matrix change for vsDeOldify
# changing range from full to limited range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="full", range_s="limited")

Enabling DeepEx:

# changing color matrix from '470bg' to '709' for vsDeOldify
clip = core.resize.Bicubic(clip, matrix_in_s="470bg", matrix_s="709")
# changing range from limited to full range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="limited", range_s="full")
# setting color range to PC (full) range.
clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)
# adjusting color space from YUV420P8 to RGB24 for vsDeOldify
clip = core.resize.Bicubic(clip=clip, format=vs.RGB24, matrix_in_s="709", range_s="full")
# adding colors using DeOldify
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="ON")
from vsdeoldify import HAVC_main
clipRef = HAVC_ddeoldify(clip=clip, deoldify_p=[0, 24, 1, 0], ddcolor_p=[1, 24, 1, 0, True])
clip = HAVC_main(clip=clip, BlackWhiteTune="light", EnableDeepEx=True, DeepExMethod=0, DeepExRefMerge=0, DeepExOnlyRefFrames=False, clip_ref=clipRef, DeepExModel=0, DeepExEncMode=0, DeepExMaxMemFrames=0)
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="OFF")
# internally changing color matrix for YUV<>RGB to '470bg' undoing color matrix change for vsDeOldify
# changing range from full to limited range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="full", range_s="limited")

Setting DeepEx Method to 5 and setting test_ref.avi as reference => no clue what to set


Switching to Custom:
(So Speed: Custom, B&W Tune: light, DeepEx enabled, DeepExMethod: 5, test_ref.avi as reference)

# changing color matrix from '470bg' to '709' for vsDeOldify
clip = core.resize.Bicubic(clip, matrix_in_s="470bg", matrix_s="709")
# changing range from limited to full range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="limited", range_s="full")
# setting color range to PC (full) range.
clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)
# adjusting color space from YUV420P8 to RGB24 for vsDeOldify
clip = core.resize.Bicubic(clip=clip, format=vs.RGB24, matrix_in_s="709", range_s="full")
# adding colors using DeOldify
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="ON")
# loading reference clip for Remaster
clipRef = core.bs.VideoSource(source="G:/TestClips&Co/files/test_ref.avi")
if clipRef.num_frames != clip.num_frames:
  raise vs.Error(f"HAVC: Reference clip ({clipRef.num_frames}) and source clip frame count ({clip.num_frames}) differ!")
if clipRef.format.id != vs.RGB24:
  refframe = clipRef.get_frame(0)
  if refframe.props.get('_ColorRange', None) != vs.RANGE_FULL :
    clipRef = core.resize.Bicubic(clipRef, range_in_s="limited", range_s="full")
  if frame.props.get('_Matrix', None) != vs.MATRIX_BT709 :
    clipRef = core.resize.Bicubic(clipRef, matrix_in_s="470bg", matrix_s="709")
  clipRef = core.resize.Bicubic(clip=clipRef, width=clip.width, height= clip.height, format=vs.RGB24, matrix_in_s="709", range_s="full")
from vsdeoldify import HAVC_deepex
clip = HAVC_deepex(clip=clip, clip_ref=clipRef, method=5, ref_merge=0, sc_framedir=None, ex_model=0, max_memory_frames=0, encode_mode=0)
# internally changing color matrix for YUV<>RGB to '470bg' undoing color matrix change for vsDeOldify
# changing range from full to limited range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="full", range_s="limited")

Enabling Chroma smoothing, Chroma Stabilizer, Darkness:

# changing color matrix from '470bg' to '709' for vsDeOldify
clip = core.resize.Bicubic(clip, matrix_in_s="470bg", matrix_s="709")
# changing range from limited to full range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="limited", range_s="full")
# setting color range to PC (full) range.
clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)
# adjusting color space from YUV420P8 to RGB24 for vsDeOldify
clip = core.resize.Bicubic(clip=clip, format=vs.RGB24, matrix_in_s="709", range_s="full")
# adding colors using DeOldify
clip = HAVC_bw_tune(clip=clip, bw_tune="light", action="ON")
# loading reference clip for Remaster
clipRef = core.bs.VideoSource(source="G:/TestClips&Co/files/test_ref.avi")
if clipRef.num_frames != clip.num_frames:
  raise vs.Error(f"HAVC: Reference clip ({clipRef.num_frames}) and source clip frame count ({clip.num_frames}) differ!")
if clipRef.format.id != vs.RGB24:
  refframe = clipRef.get_frame(0)
  if refframe.props.get('_ColorRange', None) != vs.RANGE_FULL :
    clipRef = core.resize.Bicubic(clipRef, range_in_s="limited", range_s="full")
  if frame.props.get('_Matrix', None) != vs.MATRIX_BT709 :
    clipRef = core.resize.Bicubic(clipRef, matrix_in_s="470bg", matrix_s="709")
  clipRef = core.resize.Bicubic(clip=clipRef, width=clip.width, height= clip.height, format=vs.RGB24, matrix_in_s="709", range_s="full")
from vsdeoldify import HAVC_deepex
clip = HAVC_deepex(clip=clip, clip_ref=clipRef, method=5, ref_merge=0, dark=True, smooth=True, sc_framedir=None, ex_model=0, max_memory_frames=0, encode_mode=0)
# internally changing color matrix for YUV<>RGB to '470bg' undoing color matrix change for vsDeOldify
# changing range from full to limited range for vsDeOldify
clip = core.resize.Bicubic(clip, range_in_s="full", range_s="limited")

Cu Selur