push

1 files changed, 2 insertions, 2 deletions

diff --git a/test.py b/test.py
index 4efddb8..7655cdb 100644
--- a/test.py
+++ b/test.py
@@ -66,12 +66,12 @@ if __name__ == '__main__':
                 array_a = np.multiply(im, 1/frac)
                 array_b = np.multiply(last_im, (frac-1)/frac)
                 print(im.shape, array_a.shape, array_b.shape)
-                im = np.add(array_a, array_b).clip(0, 255)
+                im = np.add(array_a, array_b)
                 last_im = tmp_im
             else:
                 last_im = im.copy()
                 print(im.shape)
-            image_pil = Image.fromarray(im)
+            image_pil = Image.fromarray(im, mode='RGB')
             image_pil.save(save_path)
             os.rename(save_path, final_path)
     webpage.save()