Feature Request: Dynamic TensorRT models and build options

# Imports

import vapoursynth as vs

# getting Vapoursynth core

import logging

import site

import sys

import os

core = vs.core

# Import scripts folder

scriptPath = 'C:/Program Files/Hybrid/64bit/vsscripts'

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

os.environ["CUDA_MODULE_LOADING"] = "LAZY"

# Force logging to std:err

logging.StreamHandler(sys.stderr)

# loading plugins

core.std.LoadPlugin(path="C:/Program Files/Hybrid/64bit/vsfilters/Support/vszip.dll")

core.std.LoadPlugin(path="C:/Program Files/Hybrid/64bit/vs-mlrt/vstrt.dll")

core.std.LoadPlugin(path="C:/Program Files/Hybrid/64bit/vsfilters/Support/fmtconv.dll")

core.std.LoadPlugin(path="C:/Program Files/Hybrid/64bit/vsfilters/SourceFilter/LSmashSource/LSMASHSource.dll")

# Import scripts

from importlib.machinery import SourceFileLoader

vsmlrt = SourceFileLoader('vsmlrt', 'C:/Program Files/Hybrid/64bit/vs-mlrt/vsmlrt.py').load_module()

from vsmlrt import Backend

import validate

# Source: 'F:\REMASTERIZACIJA\KLIPOVI ONLY BD\15.februar\davor\PRINC RIMA.AVI'

# Current color space: YUV422P8, bit depth: 8, resolution: 320x240, frame rate: 15fps, scanorder: progressive, yuv luminance scale: full, matrix: 470bg, format: JPEG

# Loading F:\REMASTERIZACIJA\KLIPOVI ONLY BD\15.februar\davor\PRINC RIMA.AVI using LWLibavSource

clip = core.lsmas.LWLibavSource(source="F:/REMASTERIZACIJA/KLIPOVI ONLY BD/15.februar/davor/PRINC RIMA.AVI", format="YUV422P8", stream_index=0, cache=0, prefer_hw=0)

frame = clip.get_frame(0)

# setting color matrix to 470bg.

clip = core.std.SetFrameProps(clip, _Matrix=vs.MATRIX_BT470_BG)

# setting color transfer (vs.TRANSFER_BT601), if it is not set.

if validate.transferIsInvalid(clip):

  clip = core.std.SetFrameProps(clip=clip, _Transfer=vs.TRANSFER_BT601)

# setting color primaries info (to vs.PRIMARIES_BT470_BG), if it is not set.

if validate.primariesIsInvalid(clip):

  clip = core.std.SetFrameProps(clip=clip, _Primaries=vs.PRIMARIES_BT470_BG)

# setting color range to PC (full) range.

clip = core.std.SetFrameProps(clip=clip, _ColorRange=vs.RANGE_FULL)

# making sure frame rate is set to 15fps

clip = core.std.AssumeFPS(clip=clip, fpsnum=15, fpsden=1)

# making sure the detected scan type is set (detected: progressive)

clip = core.std.SetFrameProps(clip=clip, _FieldBased=vs.FIELD_PROGRESSIVE) # progressive

# adjusting color space from YUV422P8 to RGBH for vsVSMLRT

clip = core.resize.Bicubic(clip=clip, format=vs.RGBH, matrix_in_s="470bg", range_in_s="full", range_s="full")

# resizing using VSMLRT, target: 640x480

clip = vsmlrt.inference([clip],network_path="D:/ONNX MODELI/2x_FeMaSR_SRX2_model_g-sim.onnx", backend=Backend.TRT(fp16=True,device_id=0,bf16=False,num_streams=1,verbose=True,use_cuda_graph=True,workspace=1073741824,builder_optimization_level=3,engine_folder="C:/Users/admin/Videos/HYBRIDMODELI",,static_shape=False,opt_shapes=[1280,1280],min_shapes=[8,8],max_shapes=[2560,2560])) # 640x480

# making sure 0-1 limits are respected

clip = core.vszip.Limiter(clip=clip, min=[0,0,0], max=[1,1,1])

# adjusting output color from: RGBH to YUV444P16 for FFV1Model

clip = core.resize.Bicubic(clip=clip, format=vs.YUV444P16, matrix_s="470bg", range_in_s="full", range_s="full") # additional resize to allow target color sampling

# set output frame rate to 15fps (progressive)

clip = core.std.AssumeFPS(clip=clip, fpsnum=15, fpsden=1)

# output

clip.set_output()