From 2235b043601e82c42a9fe55d296d50dd563221ce Mon Sep 17 00:00:00 2001 From: jules Date: Wed, 16 Jan 2019 02:21:24 +0100 Subject: a --- uprez.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/uprez.py b/uprez.py index 0be4643..249025d 100644 --- a/uprez.py +++ b/uprez.py @@ -1,18 +1,23 @@ import os import time import subprocess +import argparse start_time = time.time() parser = argparse.ArgumentParser() parser.add_argument('--folder', required=True) parser.add_argument('--out_fn', required=True) +parser.add_argument('--dataset', required=True) +opt = parser.parse_args() os.makedirs('./uprez', exist_ok=True) +dataset = opt.dataset + print(">>> {}".format(dataset)) -in_dir = os.path.abspath('./recursive/{}'.format(dataset)) +in_dir = os.path.abspath(opt.folder) 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') @@ -45,3 +50,4 @@ subprocess.call([ finish_time = time.time() print("took %02f s.".format(finish_time - start_time)) + -- cgit v1.2.3-70-g09d2