26 lines
661 B
Python
26 lines
661 B
Python
import sys
|
|
import os
|
|
|
|
sys.path.append("/workspace/tensorrt/")
|
|
import vapoursynth as vs
|
|
|
|
core = vs.core
|
|
vs_api_below4 = vs.__api_version__.api_major < 4
|
|
core.num_threads = 8
|
|
|
|
core.std.LoadPlugin(path="/usr/local/lib/libvstrt.so")
|
|
|
|
|
|
def inference_clip(video_path="", clip=None):
|
|
clip = core.bs.VideoSource(source=video_path)
|
|
|
|
clip = vs.core.resize.Bicubic(clip, format=vs.RGBH, matrix_in_s="709")
|
|
clip = core.trt.Model(
|
|
clip,
|
|
engine_path="/workspace/tensorrt/2x_AnimeJaNai_V2_Compact_36k_op18_fp16_clamp.engine",
|
|
num_streams=2,
|
|
)
|
|
clip = vs.core.resize.Bicubic(clip, format=vs.YUV420P8, matrix_s="709")
|
|
|
|
return clip
|