-
Notifications
You must be signed in to change notification settings - Fork 0
/
cain.vpy
17 lines (16 loc) · 598 Bytes
/
cain.vpy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import vapoursynth as vs
import sys
import os
import os.path
core = vs.core
core.num_threads = 8
core.std.LoadPlugin(path="/usr/local/lib/libvstrt.so")
clip = core.lsmas.LWLibavSource(r'testsrc.mp4')
clip = vs.core.resize.Bicubic(clip, format=vs.RGBS, matrix_in_s="709")
clip_pos1 = clip[1:]
clip_pos2 = clip.std.Trim(first=0,last=clip.num_frames-2)
clipstack = [clip_pos1,clip_pos2]
output = core.trt.Model(clipstack, engine_path=f"2.engine",num_streams=8)
output=core.std.Interleave([clip,output])
output = vs.core.resize.Bicubic(output, format=vs.YUV420P8, matrix_s="709")
output.set_output()