splitting image and merge them

split_and_merge.py

import numpy as np
import matplotlib.pyplot as plt

# https://stackoverflow.com/a/55176783/6660373
# 20x20 image
img = im#np.random.randint(0,9,(512,512))

# List of 4 5x20 image slices
sliced = np.split(img,4,axis=0)

# List of 4 lists of 4 5x5 image blocks
blocks = [np.split(img_slice,4,axis=1) for img_slice in sliced]

# stacking them back together
img_stacked = np.block(blocks)

# testing if the stacking works right
# print((img==img_stacked).all())

image = im
# tiles = get_tile_images(image, 128, 128)
tiles = blocks
_nrows = int(image.shape[0] / 128)
_ncols = int(image.shape[1] / 128)

fig, ax = plt.subplots(nrows=_nrows, ncols=_ncols)
# # count = 0
for i in range(_nrows):
    for j in range(_ncols):
        ax[i, j].imshow(blocks[i][j])
        ax[i, j].set_axis_off()