diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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=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('--in_dir', metavar='in_dir', default=None, help='Directory to process') parser.add_argument('--out_dir', metavar='out_dir', default='/media/blue/uprez', help='Directory to output to') |