distortion in the unpatchified image

[Mohamab29]

The original image is in shape (844,1280) and I patched it with patchify using 128 steps , because I need like +40 patches with size of (256,256) , unpatchify succeeded but sadly I have distortions in the result , is there a way to fix this ?

patched = patchify(image,(256,256),step=128)
unpatched_img = unpatchify(patched, image.shape)

[Mohamab29]

the problem was solved by making the width and height the same for example 1024 X 1024 with 128 steps
because (1024 -256) mod 128 = 0