uprez.py

1 files changed, 47 insertions, 0 deletions

diff --git a/uprez.py b/uprez.py
new file mode 100644
index 0000000..0be4643
--- /dev/null
+++ b/uprez.py
@@ -0,0 +1,47 @@
+import os
+import time
+import subprocess
+
+start_time = time.time()
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--folder', required=True)
+parser.add_argument('--out_fn', required=True)
+
+os.makedirs('./uprez', exist_ok=True)
+
+print(">>> {}".format(dataset))
+
+in_dir = os.path.abspath('./recursive/{}'.format(dataset))
+vsr_duf_inputs_dir = os.path.abspath('../vsr-duf/inputs')
+vsr_duf_results_dir = os.path.abspath('../vsr-duf/results/{}'.format(dataset))
+out_dir = os.path.abspath('./uprez')
+
+subprocess.call([
+  '/home/lens/miniconda3/envs/tf/bin/python',
+  '../vsr-duf/crop_frames.py',
+  '--in_dir', in_dir,
+  '--out_dir', vsr_duf_inputs_dir,
+  '--ratio', '6.0'
+])
+
+subprocess.call([
+  '/home/lens/miniconda3/envs/tf/bin/python',
+  '../vsr-duf/test.py',
+  '--in_dir', vsr_duf_inputs_dir,
+  '--out_dir', vsr_duf_results_dir,
+  '--network_dir', os.path.abspath('../vsr-duf'),
+])
+
+subprocess.call([
+  '/home/lens/miniconda3/envs/tf/bin/python',
+  '../vsr-duf/merge_frames.py',
+  '--in_dir', vsr_duf_results_dir,
+  '--render_dir', out_dir,
+  '--out_fn', opt.out_fn,
+  '--mov',
+])
+
+finish_time = time.time()
+
+print("took %02f s.".format(finish_time - start_time))