test

1 files changed, 6 insertions, 9 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index 83b9f81..2ffadbb 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -21,23 +21,20 @@ from time import sleep
 if __name__ == '__main__':
     opt = TestOptions().parse()
     data_opt = DatasetOptions().parse(opt.unknown)
+    opt.nThreads = 1   # test code only supports nThreads = 1
+    opt.batchSize = 1  # test code only supports batchSize = 1
+    opt.serial_batches = True  # no shuffle
+    opt.no_flip = True  # no flip
+    opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0]
 
     d = datetime.now()
     tag = "{}_{}_{}_{}".format(
         opt.name, opt.experiment, data_opt.experiment,
-        # opt.how_many, frac_a,
-        # blur, sigma, canny_lo, canny_hi,
         d.strftime('%Y%m%d%H%M'))
 
     opt.tag = tag
 
-    opt.nThreads = 1   # test code only supports nThreads = 1
-    opt.batchSize = 1  # test code only supports batchSize = 1
-    opt.serial_batches = True  # no shuffle
-    opt.no_flip = True  # no flip
-    opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0]
-
-    opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + data_opt.experiment + "/"
+    opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + tag + "/"
 
     print("create render_dir: {}".format(render_dir))
     if os.path.exists(render_dir):