Init

utils/interpolate4k.py

@@ -0,0 +1,42 @@
+import os
+import subprocess
+
+def createInterpolateScript(upscaleOut, tempName):
+    if os.path.isfile(upscaleOut + '.vpy'):
+        print('Interpolate script exists')
+        return
+
+    script = ['from vsrife import rife',
+                'import vapoursynth as vs',
+                'from vapoursynth import core',
+                'clip = core.ffms2.Source(source="./' + tempName + ' [4k][HEVC].mkv")',
+                'clip = vs.core.resize.Bicubic(clip, width=3840, height=2160, format=vs.RGBS, matrix_in_s="709")',
+                'clip = rife(clip=clip, model="4.15.lite", factor_num=2, factor_den=1)',
+                'clip = vs.core.resize.Bicubic(clip, format=vs.YUV420P8, matrix_s="709")',
+                'clip.set_output()']
+
+    if not os.path.isfile(upscaleOut + '.vpy'):
+        with open(upscaleOut + '.vpy', 'a') as fs:
+            fs.writelines([i + '\n' for i in script])
+
+def Interpolate4K(interpolateOut, upscaleOut, interpolateVideo, tempName):
+    if not interpolateVideo:
+        print('Skipped interpolation')
+        return
+
+    if os.path.isfile(interpolateOut):
+        print('Already interpolated')
+        return
+    
+    createInterpolateScript(upscaleOut, tempName)
+
+    if not os.path.isfile(upscaleOut + '.vpy'): 
+        print('=== Interpolation script not found ===')
+        return
+
+    print('Interpolating')
+    subprocess.call('vspipe -c y4m "' + upscaleOut + '.vpy" - | ffmpeg -v quiet -stats -i - -c:v hevc_nvenc -qp 5 "' + interpolateOut + '"', shell=True)
+
+    # Remove Temp Files
+    os.remove(upscaleOut + '.ffindex')
+    os.remove(upscaleOut + '.vpy')