Upload 2 files

func/rife.py

@@ -0,0 +1,33 @@
+import os
+import time
+import subprocess
+import tempfile
+import base64
+import gc
+
+from onegai.config import cfg
+
+app_name = os.path.splitext(os.path.basename(__file__))[0]
+
+def main(args):
+    return base64.b64encode(raw(args)), {}
+
+def raw(args):
+    return generate(args['exp'], args['fps'])
+
+def generate(exp, fps):
+    cd = 'cd  ./apps/ECCV2022-RIFE/;'
+    python = ' ./venv/bin/python'
+    py = './inference_video.py'
+    options = f"--exp={exp} --fps={fps} --video={f.name}"
+    f = tempfile.NamedTemporaryFile(delete=False)
+    f.close()
+    cmd = f"{cd} {python} {py} {options}"
+    subprocess.run(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+
+    f = open(f.name, 'rb')
+    content = f.read()
+    f.close()
+    os.unlink(f.name)
+
+    return content

install/rife.sh

@@ -0,0 +1,13 @@
+#!/usr/bin/bash
+
+git clone https://github.com/megvii-research/ECCV2022-RIFE rife
+cd rife
+
+python3 -m venv venv
+curl -kL https://bootstrap.pypa.io/get-pip.py | venv/bin/python
+
+venv/bin/python -m pip install -r requirements.txt
+venv/bin/python -m pip install sk-video moviepy opencv-python
+
+wget https://huggingface.co/aka7774/ECCV2022-RIFE/resolve/main/RIFE_trained_model_v3.6.zip
+unzip -o RIFE_trained_model_v3.6.zip