Merge branch 'master' of ghghgh.us:pix2pix

1 files changed, 4 insertions, 1 deletions

diff --git a/test.py b/test.py
index 62dc3cd..5b75c9b 100644
--- a/test.py
+++ b/test.py
@@ -60,9 +60,12 @@ if __name__ == '__main__':
             # image_pil.save(save_path)
             # copyfile(save_path, final_path)
             if last_im is not None:
+                frac = 4
                 tmp_im = im.copy()
-                im = np.add(np.multiply(im, 1/3), np.multiply(last_im, 2/3))
+                im = np.add(np.multiply(im, 1/(frac)), np.multiply(last_im, (frac-1)/(frac)))
                 last_im = tmp_im
+            else:
+                print(im.shape)
             image_pil = Image.fromarray(im)
             image_pil.save(save_path)
             os.rename(save_path, final_path)