lol

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')