diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/test-mogrify.py b/test-mogrify.py index c8fa7a0..7c3c84a 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -46,19 +46,22 @@ if __name__ == '__main__': model = create_model(opt) visualizer = Visualizer(opt) # create website - web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) - webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) + # web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) + # webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) # test last_im = None for i, data in enumerate(data_loader): + print(i) if i >= opt.how_many: break model.set_input(data) model.test() visuals = model.get_current_visuals() img_path = model.get_image_paths() - print('%04d: process image... %s' % (i, img_path)) - ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) + + if (i % 100) == 0: + print('%04d: process image... %s' % (i, img_path)) + # ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) im = visuals['fake_B'] tmp_path = render_dir + "frame_{:05d}_tmp.png".format(i+1) |