diff --git a/tester.py b/tester.py index cd963b3..b4a3dbf 100644 --- a/tester.py +++ b/tester.py @@ -27,8 +27,8 @@ def load_model_generator(net, epoch, opt): # ---------------------------------------- # configurations - if not os.path.exists(results_path): - os.makedirs(results_path) + if not os.path.exists(opt.results_path): + os.makedirs(opt.results_path) # Build networks generator = utils.create_generator(opt).eval() @@ -70,5 +70,5 @@ def load_model_generator(net, epoch, opt): mask = torch.cat((mask, mask, mask), 1) img_list = [second_out_wholeimg] name_list = ['second_out'] - utils.save_sample_png(sample_folder = results_path, sample_name = '%d' % (batch_idx + 1), img_list = img_list, name_list = name_list, pixel_max_cnt = 255) + utils.save_sample_png(sample_folder = opt.results_path, sample_name = '%d' % (batch_idx + 1), img_list = img_list, name_list = name_list, pixel_max_cnt = 255) print('----------------------batch_idx%d' % (batch_idx + 1) + ' has been finished----------------------')