diff options
| -rw-r--r-- | uprez.py | 47 |
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)) |