Init

utils/upcale.py

@@ -0,0 +1,38 @@
+import os
+import subprocess
+
+from utils.mediainfo import get_framerate
+
+def re_encode(input_file, upscale_out, temp_out, max_width, input_aspect):
+    if os.path.exists(upscale_out):
+        print('Skipped Pre-Encode')
+        return
+    
+    command = (
+        f'ffmpeg -v quiet -stats -i "{input_file}" '
+        '-c:v ffv1 -level 3 '
+        f'-vf "fps={get_framerate(input_file)},scale=-1:\'min({max_width},ih)\'" -aspect {input_aspect} '
+        '-pix_fmt yuv420p -color_primaries 1 -color_trc 1 -colorspace 1 '
+        '-an -sn -map_metadata -1 '
+        f'"{temp_out}"'
+    )
+
+    subprocess.call(command, shell=True)
+
+def upscale(input_file, upscale_out, max_width, input_aspect):
+    temp_out = os.path.join('1-Temp', 'source.mkv')
+    vsgan = os.path.join('utils', 'vs-realesrgan.vpy')
+
+    # Re-Encode to fix issues
+    re_encode(input_file, upscale_out, temp_out, max_width, input_aspect)
+
+    if os.path.exists(upscale_out):
+        print('Skipped Upscale')
+        return
+
+    print('Started Upscale')
+    subprocess.call(f'vspipe -c y4m {vsgan} - | ffmpeg -v quiet -stats -f yuv4mpegpipe -i - -c:v hevc_nvenc -qp 5 -aspect {input_aspect} "{upscale_out}"', shell=True)
+
+    # Remove Temp Files
+    os.remove(temp_out)
+    os.remove(temp_out + '.ffindex')