random net

1 files changed, 2 insertions, 1 deletions

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