push

1 files changed, 1 insertions, 1 deletions

diff --git a/test.py b/test.py
index 6941098..a712207 100644
--- a/test.py
+++ b/test.py
@@ -60,7 +60,7 @@ if __name__ == '__main__':
             # image_pil = ImageOps.expand(image_pil, p)
             # image_pil.save(save_path)
             # copyfile(save_path, final_path)
-            if i > 0:
+            if last_im is not None:
                 im = np.divide(im, 2)
                 tmp_im = im.copy()
                 im = np.add(im, last_im)