Merge branch 'master' of ghghgh.us:pix2pix

1 files changed, 1 insertions, 3 deletions

diff --git a/test.py b/test.py
index a712207..62dc3cd 100644
--- a/test.py
+++ b/test.py
@@ -37,7 +37,6 @@ if __name__ == '__main__':
     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
-    print(dataset.name())
     last_im = None
     for i, data in enumerate(data_loader):
         if i >= opt.how_many:
@@ -61,9 +60,8 @@ if __name__ == '__main__':
             # image_pil.save(save_path)
             # copyfile(save_path, final_path)
             if last_im is not None:
-                im = np.divide(im, 2)
                 tmp_im = im.copy()
-                im = np.add(im, last_im)
+                im = np.add(np.multiply(im, 1/3), np.multiply(last_im, 2/3))
                 last_im = tmp_im
             image_pil = Image.fromarray(im)
             image_pil.save(save_path)