From 47b11b226ac2bd97ebd81226f3658dcd9e684893 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 9 Jan 2019 23:34:28 +0100 Subject: random net --- test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test.py') diff --git a/test.py b/test.py index 673054b..b99aa56 100644 --- a/test.py +++ b/test.py @@ -7,13 +7,14 @@ import scipy import argparse import os import subprocess +from random import choice from PIL import Image from utils import LoadImage, DownSample, AVG_PSNR, depth_to_space_3D, DynFilter3D, LoadParams from nets import FR_16L, FR_28L, FR_52L parser = argparse.ArgumentParser() -parser.add_argument('--L', metavar='L', type=int, default=28, 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') -- cgit v1.2.3-70-g09d2