test

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)