From b61ef3c57798dd3ecc67ec1f4d207f0520a89f43 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 16 Jan 2019 02:34:56 +0100 Subject: lol --- crop_frames.py | 2 +- test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/crop_frames.py b/crop_frames.py index dd8c1e6..b1db08d 100644 --- a/crop_frames.py +++ b/crop_frames.py @@ -49,7 +49,7 @@ def crop_dir(): partz = fn.split('_') try: num = int(partz[-1].replace('.png', '')) - if num >= start and num <= end: + if num >= opt.start and num <= opt.end: dataset.append((raw_fn, width, height, step, side, overlap, top_offset,)) except: pass diff --git a/test.py b/test.py index 92e76e3..3847b3f 100644 --- a/test.py +++ b/test.py @@ -14,7 +14,7 @@ from utils import LoadImage, DownSample, AVG_PSNR, depth_to_space_3D, DynFilter3 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('--L', metavar='L', type=int, default=28, 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') -- cgit v1.2.3-70-g09d2