From 891ff697a091828d32dcbb7d80d1b50c4a42f379 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 15 Jan 2019 23:45:10 +0100 Subject: pre-crop --- test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index eab51b2..13c92ef 100644 --- a/test.py +++ b/test.py @@ -16,6 +16,7 @@ from nets import FR_16L, FR_28L, FR_52L parser = argparse.ArgumentParser() parser.add_argument('--L', metavar='L', type=int, default=choice([16, 28, 52]), help='Network depth: One of 16, 28, 52') parser.add_argument('--T', metavar='T', default='L', help='Input type: L(Low-resolution) or G(Ground-truth)') +parser.add_argument('--dataset', default=None, help='Name of dataset') parser.add_argument('--in_dir', metavar='in_dir', default=None, help='Directory to process') parser.add_argument('--out_dir', metavar='out_dir', default='/media/blue/uprez/results', help='Directory to output to') parser.add_argument('--network_dir', default='.', help='Path to networks') @@ -43,10 +44,10 @@ if not(args.T == 'L' or args.T =='G'): def process_dir(dir): dir_partz = dir.split('/') - dataset = dir_partz[-2] + dataset = args.dataset = dir_partz[-2] part = dir_partz[-1] tag = '_'.join([dataset, str(args.L) + 'L', part]) - out_path = os.path.join(args.out_dir, dataset, str(args.L) + 'L', part) + out_path = os.path.join(args.out_dir, dataset, part) render_path = os.path.join(args.out_dir, 'renders') os.makedirs(out_path, exist_ok=True) if args.mov_from_dirs: -- cgit v1.2.3-70-g09d2