1 files changed, 2 insertions, 2 deletions
diff --git a/test.py b/test.py
index 43a69de..9aed9bb 100644
--- a/test.py
+++ b/test.py
@@ -68,10 +68,10 @@ if __name__ == '__main__':
                 array_b = np.multiply(last_im, frac_b)
                 im = np.add(array_a, array_b).astype('int8')
                 # print(im.shape, im.dtype)
-                last_im = np.roll(tmp_im, 1, axis=0)
+                last_im = np.roll(tmp_im, 1, axis=1)
             else:
                 last_im = im.copy()
-                # print(im.shape, im.dtype)
+                print(im.shape, im.dtype)
             image_pil = Image.fromarray(im, mode='RGB')
             image_pil.save(save_path)
             os.rename(save_path, final_path)