diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 15 |
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): |